
Posted 4 months ago
Distributed Machine Learning Engineer
Sunnyvale, CAOn-siteFull-time
AI Summary
A Distributed Machine Learning Engineer optimizes performance of ML software stacks for training and inference, designs benchmarks and tooling, troubleshoots bottlenecks, and contributes to system development for cutting‑edge foundation model research.
About this role
About the Institute of Foundation Models
We are a dedicated research lab for building, understanding, using, and risk-managing foundation models. Our mandate is to advance research, nurture the next generation of AI builders, and drive transformative contributions to a knowledge-driven economy.
As part of our team, you’ll have the opportunity to work on the core of cutting-edge foundation model training, alongside world-class researchers, data scientists, and engineers, tackling the most fundamental and impactful challenges in AI development. You will participate in the development of groundbreaking AI solutions that have the potential to reshape entire industries. Strategic and innovative problem-solving skills will be instrumental in establishing MBZUAI as a global hub for high-performance computing in deep learning, driving impactful discoveries that inspire the next generation of AI pioneers.
The Role
The Distributed ML Engineer will play a role at the forefront of optimizing performance for the machine learning software stacks, especially at training and inference, and support the team to develop new and cutting-edge systems. The ideal candidate will have a strong background in parallel computing, and hands-on experience in system level coding, debug methodologies, and large-scale machine learning experience.
Key Responsibilities
Academic Qualifications
Skills
DebuggingGPU UtilizationHPCKernel-developmentLarge-scale Machine LearningParallel ComputingPerformance BenchmarksSystem Level Coding
Explore related jobs
More jobs at Institute of Foundation Models
Developer Relations & Community Manager – Institute of Foundation Models (IFM)Sunnyvale, CA
Website Developer & Visual DesignerSunnyvale, CA
AI Engineer Internship – LLM DataAbu Dhabi
Research Scientist – World Modeling, DataSunnyvale, CA
Social Media ManagerSunnyvale, CA
Content Marketing & Editorial, Sr. ManagerSunnyvale, CA
Browse these categories
Market data for ai / ml engineer roles
All reports →- SeriesRole reportsOne role family at a time: how many openings, what changed this week, who is hiring, what it pays.
- SeriesSalary reportsWhat employers publish in job postings, by level and workplace. Not self-reported pay.
- Market overviewState of tech hiring, September 2026: up 4.8%Tech hiring rose 4.8% month over month in September 2026, with 411,122 new listings. Customer support and account executive roles led the growth.