Institute for Foundation Models

Staff Scientist, World Model

Sunnyvale, CA
USD 300k - 600k
Python LLM Diffusion models Reinforcement Learning
Description

Staff Scientist - World Model

Team: Research

Location: Sunnyvale, CA

Commitment: Full-time

Workplace Type: onsite

Salary:

Total compensation target: $300,000.00 – $600,000.00 per annum (inclusive of base salary and target bonus up to 30% of base salary), commensurate with experience.

Visa Sponsorship
This position is eligible for visa sponsorship.

Benefits Include
*Comprehensive medical, dental, and vision benefits 
 *Bonus
*401K Plan
*Generous paid time off, sick leave and holidays
*Paid Parental Leave
*Employee Assistance Program
*Life insurance and disability


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
You will own the end‐to‐end strategy, design and delivery of our general-purpose world modeling efforts. You’ll translate cutting-edge research (e.g., the PAN framework) into robust, production-ready simulators, guide a multidisciplinary team of engineers and scientists, and ensure alignment with IFM’s mission.
 
Key Responsibilities
Technical Leadership & Vision
Define and evolve the overall world model architecture, drawing on the PAN principles:
Multimodal data ingestionMixed continuous/discrete representationsHierarchical generative modeling with an enhanced LLM backbone and diffusion-based predictorsGenerative loss grounded in real observationsSimulation for RL-based agent training 
Establish performance, safety and evaluation benchmarks, driving continuous improvement.
Project & Team Management
Oversee planning, resourcing, and timeline for world model projects. Manage research engineers and scientists (e.g., data curators, RL experts, simulator devs) to achieve unified progress.
Cross-Functional Collaboration
Partner with agent, reasoning, and deployment teams to integrate world model outputs into downstream applications (robotics, multi-turn dialogue, autonomous systems).Liaise with external collaborators (academia & industry) to incorporate the latest advances and tooling.
Governance & Communication
Report project status, risks, and key insights to senior leadership and stakeholders.Champion best practices in 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.
Proven 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 experience with reinforcement learning frameworks (policy learning, planning with latent dynamics).
Strong leadership skills: 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.
Track record of taking research prototypes into production systems.
Institute for Foundation Models
Institute for Foundation Models

0 applies

0 views

There are more than 50,000 engineering jobs:

Subscribe to membership and unlock all jobs

Engineering Jobs

60,000+ jobs from 4,500+ well-funded companies

Updated Daily

New jobs are added every day as companies post them

Refined Search

Use filters like skill, location, etc to narrow results

Become a member

🥳🥳🥳 452 happy customers and counting...

Overall, over 80% of customers chose to renew their subscriptions after the initial sign-up.

To try it out

For active job seekers

For those who are passive looking

Cancel anytime

Frequently Asked Questions

  • We prioritize job seekers as our customers, unlike bigger job sites, by charging a small fee to provide them with curated access to the best companies and up-to-date jobs. This focus allows us to deliver a more personalized and effective job search experience.
  • We've got over 200,000 jobs from 15,000+ vetted companies. No fake or sleazy jobs here!
  • We aggregate jobs from 15,000+ companies' career pages, so you can be sure that you're getting the most up-to-date and relevant jobs.
  • We're the only job board *for* software engineers, *by* software engineers… in case you needed a reminder! We add thousands of new jobs daily and offer powerful search filters just for you. 🛠️
  • Every single hour! We add 2,000-3,000 new jobs daily, so you'll always have fresh opportunities. 🚀
  • Typically, job searches take 3-6 months. EchoJobs helps you spend more time applying and less time hunting. 🎯
  • Check daily! We're always updating with new jobs. Set up job alerts for even quicker access. 📅

What Fellow Engineers Say