Research Engineer - World Modeling in Sunnyvale

Energy Jobline is the largest and fastest growing global Energy Job Board and Energy Hub. We have an audience reach of over 7 million energy professionals, 400,000+ monthly advertised global energy and engineering jobs, and work with the leading energy companies worldwide.

We focus on the Oil & Gas, Renewables, Engineering, Power, and Nuclear markets as well as emerging technologies in EV, Battery, and Fusion. We are committed to ensuring that we offer the most exciting career opportunities from around the world for our jobseekers.

Job DescriptionJob DescriptionAbout the Institute of Foundation ModelsWe are a dedicated research lab for building, understanding, using, and risk-managing foundation models. Our mandate is to advance research, nurture the next 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 of AI pioneers.


The Role
We are the AllWorld Team under the Institute of Foundation Model (IFM) at MBZUAI. At AllWorld, we are pioneering the development of the PAN (Physical, Agentic, and Networked) world models—the next- foundation models to unlock machine intelligence beyond lingual. 
Our mission is to tackle the fundamental challenges of world modeling and establish a new paradigm for next- machine reasoning. We are looking for passionate individuals who share our vision and are eager to push the boundaries of AI together. Key Responsibilities: Data Infrastructure & Pipelines

• Design, implement, and maintain scalable video data pipelines to support large-scale training. 
• Develop data preprocessing, transformation, and synthesis workflows to support world model training. 
• Contribute to building high-quality data annotation pipelines to ensure accurate and consistent labels across large-scale datasets. 

Key Responsibilities: Training & Inference Systems

• Support the training of multimodal foundation models (e.g., video diffusion models, world models) by developing and optimizing distributed training systems. 
• Improve inference and serving efficiency for real-time interaction through model optimization and system tuning. 
• Monitor system health and performance, and contribute to debugging and optimization at scale. 

Key Responsibilities: Collaboration & Integration

• Work closely with research teams to understand experimental goals and translate ideas into reliable and maintainable infrastructure and tools. 
• Integrate novel research prototypes into production-ready systems and ensure reproducibility at scale. 
• Participate in design and code reviews, ensuring code quality, efficiency, and compliance with best practices. 

Key Responsibilities: Benchmarking & Evaluation

• Contribute to the development of tools and infrastructure to evaluate model performance using rigorous quantitative benchmarks, including metrics for physical accuracy and controllability. 

Key Responsibilities: Codebase & Documentation

• Maintain and extend shared codebases, contribute to internal documentation, and support onboarding of new team members or collaborators. 
• Write clean, efficient, and well-tested code for components across the model development lifecycle. 

Key Responsibilities

• Support contributions to research papers and demos when engineering work plays a significant role. 
• Help represent the team’s engineering excellence in internal and external forums when appropriate. 

Academic Qualifications

• MSc or PhD in Machine Learning or Computer Science, or equivalent industry experience. 

Professional Experience Required

• Proficient in data collection, cleaning, and transformation at scale, including designing robust pipelines for multimodal datasets (e.g., video, audio, text). 
• Practical experience with web scraping and crawling frameworks (e.g., scrapy, selenium, playwright, BeautifulSoup) to collect and curate high-quality web-scale datasets. 
• Experience in large-scale model training (LLMs or Diffusion Models) on large clusters. 
• Hands-on experience with state-of-the-art video generative models (e.g., Sora, Veo2, MovieGen, CogVideoX, etc.). 
• Experiences in building and optimizing large-scale video data pipelines. 
• Experience in accelerating diffusion model inference for improved efficiency. 
• Exceptional problem-solving and troubleshooting skills to tackle complex technical challenges. 
• Strong systems and engineering expertise in deep learning frameworks such as PyTorch. 
• Strong communication and collaboration skills for effective cross-functional teamwork. 
• Demonstrated ability to solve complex system-level challenges and debug failures across the training/inference stack (e.g., memory issues, deadlocks, I/O bottlenecks).

Visa SponsorshipThis 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

If you are interested in applying for this job please press the Apply Button and follow the application process. Energy Jobline wishes you the very best of luck in your next career move.