Principal ML Investigator

The role

A principal investigator will partner with ML leaders to formulate a new effort and build a new team and capabilities. The role involves adapting algorithms to the Cerebras platform and collaborating with other teams to co-design hardware and software architectures.

What you will do

• Build up a team capable of industry research and advanced development.
• Organize various advanced development topics into cohesive agenda.
• Adapt novel algorithms and model architectures to run on the Cerebras platform.
• Systematically train, tune, and evaluate models to guide/advise production scenarios.
• Collaborate with other teams to co-design next-generation hardware and software architectures.
• Collaborate with external partners to drive insight and credibility.

Required qualifications

• PhD in Computer Science or related field.
• Strong grasp of ML theory in one or more of the above areas.
• Proven experience engineering ML systems for scale or production deployment.
• Experience leading a team of researchers or engineers.

Nice to have

• 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 to target hardware, 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.

About the company

Cerebras Systems builds the world's largest AI chip. The company works with leading model labs, global enterprises, and cutting-edge AI-native startups.