World Model Lead
Role Overview
Own end-to-end strategy, design, and delivery of general-purpose world modeling.
Translate research, including the PAN framework, into robust, production-ready simulators.
Key Skills for This Role
Full Job Posting
About the institute
The Institute of Foundation Models is a research lab focused on building, understanding, using, and risk-managing foundation models.
Its mandate includes advancing research, developing AI builders, and contributing to a knowledge-driven economy.
Role
Own end-to-end strategy, design, and delivery of general-purpose world modeling.
Translate research, including the PAN framework, into robust, production-ready simulators.
Technical leadership
- Define architecture covering multimodal ingestion, mixed continuous/discrete representations, hierarchical generative modeling, an enhanced LLM backbone, and diffusion-based predictors.
- Ground generative loss in real observations and use simulation for reinforcement-learning-based agent training.
- Set performance, safety, and evaluation benchmarks and drive continuous improvement.
Projects and collaboration
- Oversee project planning, resourcing, and timelines; manage research engineers and scientists.
- Integrate world-model outputs with agent, reasoning, and deployment teams for robotics, multi-turn dialogue, and autonomous systems.
- Work with academic and industry collaborators and report project status, risks, and insights to stakeholders.
- Champion reproducibility, documentation, and knowledge sharing.
Required qualifications
- Ph.D. or M.S. with 8+ years in AI research or engineering specializing in world modeling, simulation, or generative modeling.
- Track record building large-scale simulators or predictive models for complex environments.
- Deep expertise in transformer-based LLMs, diffusion models, and hierarchical latent representations.
- Hands-on reinforcement-learning framework experience, including policy learning and planning with latent dynamics.
- Leadership skills in project management, cross-site coordination, and team mentorship.
Preferred qualifications
- Experience leading multi-location technical teams in fast-paced R&D settings.
- Published contributions to world-model architectures or simulation benchmarks.
- Experience moving research prototypes into production systems.
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