Machine Learning Engineer

MBZUAI (Mohamed bin Zayed University of Artificial Intelligence)
Department Icon IT / Software Development & Related
Posted: 1 week ago
0-15 years
Abu Dhabi, UAE
IT / Software Development & Related
full time
Full Time
Closes 2 Jan 2027
Posted: 1 week ago
Let Jobaaj One apply for you

An AI agent that applies to roles like this on employers' own sites. 25 applications free, no card needed.

Job Description
About Company
Similar Jobs
Please verify your account first! Send OTP

Job Description

Job description / Role
Job Type
Full Time
Job Location
Abu Dhabi, UAE
Nationality
Any Nationality
Salary
Not Specified
Gender
Not Specified
Arabic Fluency
Not Specified
Job Function
IT - Software & Web Development
Company Industry
Software & Internet Services
About the institute of foundation models

The Institute of Foundation Models (IFM) at MBZUAI is a research lab dedicated to meaningful foundation model research — building models from scratch, understanding them deeply, and publishing work that shapes the field. You’ll work alongside world-class researchers and engineers on problems that directly define the models we ship.

The role

Join the PAN world model project — our effort to build world models: foundation models that simulate, predict, and interact with the physical world. As a machine learning engineer, you’ll own the engineering backbone of PAN: large-scale video and simulation data pipelines, distributed training for diffusion transformers, game-engine simulation environments, and world-model integration into robotics — turning research ideas into reliable, scalable systems.

What youll do
  • Build and maintain large-scale video and simulation data pipelines — collection, cleaning, annotation, and filtering — to support world model training.
  • Develop and optimize distributed training systems for large-scale diffusion transformers and world models.
  • Build interactive simulation environments (e.g., Unreal Engine, Blueprint-based gyms, game integrations) for training and evaluating world models.
  • Integrate world action models/video action models into robotics systems.
  • Optimize inference and serving for real-time interaction, and turn research prototypes into reliable, reproducible systems.
    • What were 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 (e.g., Cosmos-3, LTX 2.3, Self-Forcing, Lingbot-World, or comparable systems).
  • Deep expertise in at least one of the following areas:
    • Full-stack data pipelines — large-scale video data pipelines and/or simulation data collection; annotation and filtering workflows for video/world model training.
    • Model training and infrastructure — training large-scale diffusion transformers on large GPU clusters.
    • Rendering engines and simulation — Unreal Engine and Blueprint-based gym environments, game-engine integration, building interactive simulated environments.
    • World action models and robotics — world action models/video action models, action-conditioned video generation, world-model applications in robotics.
  • Strong engineering expertise in deep learning frameworks such as PyTorch, with the ability to debug failures across the training/inference stack (memory issues, deadlocks, I/O bottlenecks).
  • Highly proficient with modern AI coding agents and web-based coding tools (e.g., Claude Code, Codex, Cursor), and skilled at leveraging them to dramatically accelerate engineering workflows.
    • Nice to have
  • Experience accelerating diffusion model inference (distillation, few-step generation, real-time interactive generation).
  • Practical experience with web scraping and crawling frameworks (e.g., Scrapy, Playwright, Selenium) for web-scale data curation.
  • Experience with Unreal Engine C++/Blueprint development or other game-engine programming.
  • Experience deploying world models in robotics or embodied-AI settings.

    • We may use artificial intelligence (AI) tools to support parts of the hiring process, such as reviewing applications, analyzing resumes, or assessing responses and identifying potential inconsistencies or verification signals in application materials based on available information. These tools assist our recruitment team but do not replace human judgment. Final hiring decisions are ultimately made by humans.

    Skills

    CRoboticsWeb Development

    If a job posting appears fraudulent, asks for payment, contains misleading information, or violates our guidelines, please report it immediately. Our team will review it promptly, Jobaaj does not charge any fee from the applicants.

    About Company

    MBZUAI is the world's first graduate-level, research-based university dedicated to artificial intelligence. It offers Master's and PhD programs.

    Similar Jobs

    View all
    Bag Logo
    Jobaaj
    Get jobs like this by email

    New IT / Software Development & Related roles in Abu Dhabi.

    Job Alert
    Google hiring for Specific Roles Apply Now!
    1 min ago
    New Opportunity
    Amazon is hiring freshers Apply Now!
    5 min ago
    Featured Jobs
    Microsoft opening 50+ positions Apply Now!
    10 min ago

    One email a day. Unsubscribe any time.