Principal ML Investigator
Sunnyvale, CA • FullTime
Posted 10mo ago
Job Location
Sunnyvale, CA
Tech Stack
Remote Work Policy
On-site
Employment Type
FullTime
Categories
Machine Learning Engineer
About the job
Cerebras Systems is seeking a Principal Investigator to lead a new Machine Learning (ML) team focused on innovative AI efforts. This role involves partnering with ML leaders to define the team's direction, build its capabilities, and coordinate with existing ML teams (Field ML, Applied ML, Core ML). The new team will explore critical areas such as post-training and reinforcement learning, dataset curation and optimization, LLM pretraining, sparsity techniques, and various AI domains including coding agents and generative language models.
Responsibilities
- Build a team capable of industry research and advanced development.
- Organize advanced development topics into a cohesive agenda.
- Adapt novel algorithms and model architectures for the Cerebras platform.
- Systematically train, tune, and evaluate models for production scenarios.
- Collaborate with other teams on next-generation hardware and software architectures.
- Engage with external partners (customers, academia) to drive insights and credibility.
Requirements
- PhD in Computer Science or related field.
- Strong grasp of ML theory in relevant areas.
- Proven experience engineering ML systems for scale or production deployment.
- Experience leading a team of researchers or engineers.
- Track record of patents or publications in top-tier conferences or journals.
- Experience with large language models (e.g., GPT family, Llama).
- Experience with distributed training concepts and frameworks.
- Experience in training speed optimizations, such as model architecture transformations or low-level kernel development (e.g., Triton).
- Ability to analytically model or optimize system performance.
Benefits
- Build a breakthrough AI platform beyond the constraints of the GPU.
- Publish and open source cutting-edge AI research.
- Work on one of the fastest AI supercomputers in the world.
- Job stability with startup vitality.
- Simple, non-corporate work culture that respects individual beliefs.