Applied Machine Learning Research Scientist

About the role

Cerebras Systems builds the world's largest AI chip, 56 times larger than GPUs. This architecture allows Cerebras to deliver industry-leading training and inference speeds; over 10 times faster than GPU-based hyperscale cloud inference services. This order of magnitude increase in speed is transforming the user experience of AI applications, unlocking real-time iteration and increasing intelligence via additional agentic computation. Cerebras works with the leading model labs, global enterprises, and cutting-edge AI-native startups. OpenAI recently announced a multi-year partnership with Cerebras, to deploy 750 megawatts of scale, transforming key workloads with ultra high-speed inference.

Responsibilities

• Apply post-training techniques (e.g. RLVR, RLHF, GRPO etc.) techniques to improve model performance.
• Build and maintain evaluation pipelines to measure model performance across tasks and domains.
• Debug issues across the ML stack, including data pipelines, training jobs, model outputs and mixed or lower precision computation.
• Collaborate with researchers to translate ML ideas into efficient, scalable implementation.
• Design, implement, and scale ML pipelines across all stages of LLM development (pretraining, fine-tuning, alignment).
• Work with large datasets, including dataset generation, filtering, and synthetic data approaches.
• Optimize training and inference workflows for performance, efficiency, and reliability.
• Contribute high-quality, maintainable code to shared ML infrastructure.

Qualifications

• Bachelor’s or Master’s degree in Computer Science, Engineering, or a related field.
• 4+ years of experience (including internships, research, or industry experience) working with machine learning systems; we are hiring multiple positions for various levels.
• Strong programming skills in Python.
• Experience with ML frameworks such as PyTorch.
• Solid understanding of machine learning fundamentals.
• Familiarity with deep learning architectures, particularly transformers.
• Ability to read and understand modern ML papers and implement key ideas.

Preferred Skills & Qualifications

• Experience working with large language models (training, fine-tuning, and evaluation).
• Familiarity with reinforcement learning concepts.
• Experience with distributed training frameworks (e.g., FSDP, Megatron).
• Experience working with large-scale datasets and data pipelines.
• Experience debugging or optimizing ML systems for performance.
• Contributions to meaningful codebases, projects, or open-source systems.