Machine Learning Engineer, World Models and ML Infrastructure

Job overview

The Machine Learning Engineer, World Models and ML Infrastructure role is onsite in Sunnyvale, CA. The stated salary range is $150,000–$450,000 per year. The role supports the AllWorld Team at the Institute of Foundation Models, which develops physical, agentic, and networked world models. This engineer builds and operates scalable, reliable infrastructure and MLOps workflows for research data pipelines, experimentation, and evaluation. The work helps researchers translate experimental workflows into production-ready systems while supporting the institute’s foundation-model research.

What you'll do

• Design and operate scalable AWS ML infrastructure
• develop MLOps workflows for data versioning
• manage distributed data-processing and model-evaluation systems
• make infrastructure architecture decisions
• implement monitoring, logging, and alerting
• integrate data-quality user interfaces
• maintain experiment-tracking and system-health dashboards
• partner with researchers.

What we're looking for

Skills & competencies

machine learning engineer

world models

ml infrastructure

mlops

aws

distributed systems

sunnyvale ca

institute of foundation models

$150k–$450k/year

visa sponsorship available

medical insurance

401(k)

paid parental leave

Work arrangement

• Weekend coverage required

Benefits & perks

• Medical, dental, and vision benefits
• bonus
• 401(k)
• paid time off, sick leave, and holidays
• paid parental leave
• employee assistance program
• life insurance and disability coverage.

Why this role

Visa sponsorship is available. Benefits include medical, dental, and vision coverage, bonus, 401(k), paid time off, sick leave, holidays, paid parental leave, employee assistance, life insurance, and disability coverage.

About the employer

Institute of Foundation Models is hiring for this role. Industry: Research and Development in the Social Sciences and Humanities. Sector: 54.