Staff Product Manager, ML Platform

About The Role

In this role, you will design and own the primary interface that ML researchers and scientists use to train massive LLMs (1+ trillion parameters) on Cerebras chips. Speed of iteration is critical for accelerating breakthroughs, and you’ll be at the helm of ensuring we systematically reduce the “time-to-insight” for ML research across diverse domains. 

You will work with a deeply technical product team to drive the vision and strategy for Cerebras’ ML training ecosystem, including CSTorch (our PyTorch-equivalent framework) and Model Zoo (a library of high-level abstractions and domain-specific tools for training and fine-tuning LLMs). You will lead the creation of a seamless platform that enables researchers to preprocess data, pre-train, fine-tune, and evaluate models effortlessly on Cerebras hardware. By building intuitive workflows, extensible tools, and integrated libraries, you’ll empower both cutting-edge ML research and domain-specific innovation. 

As the Cerebras ML Platform PM, you’ll play a pivotal role in advancing AI across industries, working with the most cutting-edge training techniques and collaborating with a world-class research and engineering team. 

Responsibilities

• Develop and provide a deep intuition for the ML researcher training workflow, including data preprocessing, training, fine-tuning, and evaluations.  
• Define and execute the product roadmap for Model Zoo (our ML training library) and CSTorch (our ML framework), ensuring they form a flexible, beautifully designed, and extensible platform. 
• Collaborate with our internal AppliedML team, as well as external ML researchers and domain scientists to design features that dramatically reduce time-to-insight and accelerate breakthroughs. 
• Drive cross-functional collaboration to align product roadmaps and execute priorities across frameworks and libraries. 
• Be the voice of the user! Define relevant success metrics and continuously incorporate both feedback and emerging trends in ML to refine CSTorch and ModelZoo, maintaining leadership in the space. 
• Work across Product, Engineering, and business leadership to help define our product go-to-market approach to maximize value to users and expand our user community over time. 
• Communicate roadmaps, priorities, experiments, and decisions clearly across a wide spectrum of audiences from internal customers to executives. 

Skills And Qualifications

• Bachelor’s or Master’s degree in computer science, electrical engineering, physics, mathematics, a related scientific/engineering discipline, or equivalent practical experience.
• 4-6+ years of product management experience in developer tools, ML frameworks, or software platforms. 
• Strong understanding of typical training and fine-tuning workflows, including model development, iterative experimentation, and debugging. 
• Familiarity with machine learning/deep learning concepts and techniques for training modern models.
• Proven ability to collaborate across engineering, research, and user-facing teams to deliver impactful solutions. 
• Experience working with a data science/ML stack, including TensorFlow and PyTorch.
• Experience developing machine learning applications or building tools for machine learning application developers.
• An entrepreneurial sense of ownership of overall team and product success, and the ability to make things happen around you. A bias towards getting things done, owning the solution, and driving problems to resolution.
• Outstanding presentation skills with a strong command of verbal and written communication.