Unlock your potential at Groq as we seek a Machine Learning Engineer with a focus on draft model optimization and advanced training methodologies. Become part of a mission-driven company.

As a Machine Learning Engineer in our advanced model development team, you will specialize in pre-training and post-training techniques while enhancing models for speculative decoding. The ideal candidate will have substantial experience in quantization-aware training and practical skills in optimizing transformer-based architectures for performance.

Key Responsibilities: • Drive pre-training and post-training for AI models • Optimize pipelines for low-precision while retaining accuracy • Facilitate collaboration across various technical teams • Develop tools for assessing training success • Contribute to exploratory training design

Requirements: • At least 5 years of experience in the ML domain • Proven skills with transformer-based frameworks • Expertise in Python and ML p...