Applied Machine Learning Research Scientist
Cerebras Systems, Inc.Visit Cerebras Systems, Inc. website
Cerebras builds wafer-scale AI computing systems and a cloud inference platform for training, fine-tuning, and serving AI models.
Sunnyvale, California, United States
About Cerebras Systems, Inc.
Cerebras Systems is an AI-infrastructure company founded in 2015. It sells rack-scale wafer-scale computing systems and provides cloud-based, API-accessible AI inference alongside on-premises deployments.
Skills
About the Role
You will implement scalable ML workflows for LLM pretraining, fine-tuning, and reinforcement learning post-training. You will build evaluation and data pipelines, debug issues across the ML stack, optimize training and inference, and contribute maintainable infrastructure code.
Requirements
- Machine learning
- Python
- PyTorch
- deep learning
- transformer
- ML paper implementation
Responsibilities
- Apply post-training techniques to improve model performance
- Build and maintain model evaluation pipelines
- Debug data pipelines, training jobs, model outputs, and lower-precision computation
- Translate ML ideas into scalable implementations
- Design and scale ML pipelines for pretraining, fine-tuning, and alignment
- Generate, filter, and use large datasets and synthetic data
- Optimize training and inference workflows for performance, efficiency, and reliability
- Contribute maintainable code to shared ML infrastructure