Machine Learning Engineer – World Modeling
Role Overview
Join the PAN world-model project, building models that simulate, predict, and interact with the physical world.
Own the engineering backbone across data pipelines, distributed training, simulation environments, and robotics integration.
Key Skills for This Role
Full Job Posting
About the Institute
The Institute of Foundation Models at MBZUAI is a research lab focused on foundation model research, building models from scratch, understanding them, and publishing research.
The role
Join the PAN world-model project, building models that simulate, predict, and interact with the physical world.
Own the engineering backbone across data pipelines, distributed training, simulation environments, and robotics integration.
What you'll do
- Build and maintain video and simulation data pipelines for collection, cleaning, annotation, and filtering.
- Develop and optimize distributed training systems for diffusion transformers and world models.
- Build interactive simulation environments using tools such as Unreal Engine, Blueprint-based gyms, and game integrations.
- Integrate world action models and video action models into robotics systems.
- Optimize inference and serving for real-time interaction and make research prototypes reliable and reproducible.
What we're looking for
- BSc or above in Machine Learning, Computer Science, Robotics, or a related field, or equivalent industry experience.
- Hands-on experience with state-of-the-art video generative models and world models, or comparable systems.
- Deep expertise in at least one of data pipelines, model training and infrastructure, rendering and simulation, or world action models and robotics.
- Strong deep-learning framework expertise, such as PyTorch, and ability to debug memory, deadlock, and I/O issues across training and inference.
- High proficiency with modern AI coding agents and web-based coding tools, such as Claude Code, Codex, and Cursor.
Nice to have
- Experience accelerating diffusion inference through distillation, few-step generation, or real-time interactive generation.
- Web scraping and crawling experience for web-scale data curation.
- Unreal Engine C++/Blueprint development or other game-engine programming experience.
- Experience deploying world models in robotics or embodied-AI settings.
Apply for this job in 1 click
Skip the repetitive application forms
Install the Base Career Chrome Extension and autofill job applications across major job boards with your profile.
Trusted by over 500,000 job seekers on Base Career
More from this employer
More jobs at Institute of Foundation Models
Machine Learning Engineer — Pre-training (LLM)
Sunnyvale, USA
Machine Learning Engineer — Reinforcement Learning
Sunnyvale, USA
High Performance Computing Software Engineer - Supercomputing
Abu Dhabi, UAE
The Institute of Foundation Models is seeking an HPC software engineer to build and operate compute infrastructure for large-scale AI workloads. The role covers distributed software, low-level performance optimization, c
Research Scientist - World Modeling
Abu Dhabi, UAE
The Institute of Foundation Models seeks a Research Scientist to lead research for its PAN world model project, spanning video generation, action-conditioned models, robotics, and embodied AI. The role includes developin
Research Engineer - NLP
Abu Dhabi, UAE
The Institute of Foundation Models is seeking an NLP research engineer to develop and improve efficient large language models and related AI systems. The work includes research, algorithm implementation, model integratio
Data & Eval Operations Program Manager
Sunnyvale, USA
Machine Learning Engineer — GPU Kernel
Sunnyvale, USA
Machine Learning Engineer — Pre-training (LLM)
Sunnyvale, USA
Machine Learning Engineer — Reinforcement Learning
Sunnyvale, USA
HPC Engineer
Sunnyvale, USA
High Performance Computing Software Engineer - Supercomputing
Abu Dhabi, UAE
Research Scientist - World Modeling
Abu Dhabi, UAE
Research Engineer - NLP
Abu Dhabi, UAE