Machine Learning Engineer - World Model
Institute of Foundation Models
Job Description
Job Description
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 Team
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-generation 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-generation machine reasoning. We are looking for passionate individuals who share our vision and are eager to push the boundaries of AI together.
Role Overview
We’re looking for a Machine Learning Engineer focused on ML infrastructure and MLOps to design and operate the systems that power our research environment. You’ll build scalable, reliable, and observable cloud infrastructure, working closely with researchers to support data pipelines, experimentation, and evaluation workflows.
This role balances fast-moving research needs with production-grade systems, ensuring that experimental work can scale reliably when needed.
Key Responsibilities
Design, build, and operate scalable ML infrastructure on AWS (e.g., compute, storage, networking, access control).
Develop and maintain MLOps workflows for data versioning
Build and manage distributed systems for large-scale data processing (filtering, captioning, etc.) and model evaluation.
Own architecture decisions for ML infrastructure and drive best practices in reliability, scalability, and cost efficiency.
Implement observability across systems, including monitoring, logging, and alerting.
Integrate OpenWebUI, Gradio, or similar UIs for data quality assurance
Build and maintain dashboards for experiment tracking and system health.
Partner closely with researchers to translate experimental workflows into robust, scalable systems.
Qualifications
Must-Haves
3+ years of experience in MLOps, ML infrastructure, or related backend/platform engineering roles.
Strong experience with cloud platforms (preferably AWS) and core services for compute, storage, and access control.
Experience designing and operating distributed systems (e.g., Kubernetes, Ray, or similar frameworks).
Solid software engineering skills, including system design, debugging, and testing (Python, Docker, Git).
Familiarity with data processing and pipeline orchestration tools (e.g., Spark, Kafka, or similar).
Experience with observability practices (monitoring, logging, alerting).
Ability to work closely with researchers and translate ambiguous requirements into production-ready systems.
Nice-to-Haves
Experience in fast-paced or research-driven environments.
Experience with large-scale video or multimodal data pipelines.
Experience building automated model evaluation or benchmarking systems.
Knowledge of cost optimization, security, and networking in multi-tenant environments.
Familiarity with modern developer and AI-assisted coding (e.g., Codex, Cursor, Claude Code)
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
$174.72k - $295.68k
...through cutting-edge R&D in AI, machine learning, and smart connectivity.We are looking... ...a full-time Machine Learning Engineer / Research Scientist to drive the modeling and algorithmic development of... ....You will work closely with world-class researchers, perception and...SuggestedFull time$272k - $431.25k
...FlashDreams are the foundation for a new generation of interactive world-model systems. With this release, NVIDIA has set the standard for... ...model release with an associated open source inferencing engine. We are on a mission to keep raising that bar, making interactive...SuggestedFull time$190k - $250k
...U.S. Department of Defense.Kodiak is building AI that doesn't just perceive the world, it learns how the physics of the world works. We are developing large-scale generative world models that learn to predict realistic, physically consistent futures from real-world sensor...SuggestedTemporary workWork at officeVisa sponsorship$39 - $66 per hour
...Valley focuses on Foundation Models, Big Data Visual Analytics, Explainable... ..., Data Science, AI System Engineering, Time-series Analysis. We... ...As a Research Intern for World-Action Models / VLA for Autonomous... ...and proficiency with deep learning frameworks. Experience...SuggestedWork experience placementInternshipLocal areaWorldwideBosch USA
Sunnyvale, CA1 day ago$165k - $185k
...Valley focuses on Foundation Models, Big Data Visual Analytics,... ...Robotics, Data Science, AI System Engineering, Time-series Analysis. We... .... By fusing cutting-edge machine learning, data analysis, and interactive... ...Bosch teams around the world. Their creativity is the key...SuggestedWork experience placementWorldwide$272k - $431.25k
...the future, we’re generating it! Our world model team is pushing the boundaries of multimodal... ...to see:Strong research background in machine learning, computer vision, multimodal AI,... ...experience in Computer Science, Electrical Engineering, Robotics, Machine Learning, AI, or a...Full time- ...intelligence to every moving machine on the planet. Applied... ...Bangalore; Seoul; and Tokyo. Learn more at applied.co. We are... ...research on pretraining world-action foundation model with various world modalities... ...the Research Scientists and Engineers on high quality research...For contractorsFor subcontractorCasual workInternshipWork at officeImmediate startRemote workDay shift
Applied Intution
Sunnyvale, CA4 days ago $184k - $287.5k
...cars that can understand the world. Doing what’s never been... ...creative and expert Systems Engineer to help us expand NVIDIA's autonomous... ...e.g. error budget creation, modeling of scenarios), data analysis... ...autonomous vehicles and/or machine learning, computer visionYou have...Full time$117.7k - $221.4k
...value data from large-scale real-world sensor streams. The team sits at the intersection of machine learning, data infrastructure, and... ...depends not only on stronger models, but also on better infrastructure... ...model reflects how Cola engineers think: build durable intermediate...Full timeLocal areaRemote workWork from homeRelocation packageFlexible hours$158.3k - $297k
Business UnitTechnology Engineering Group (TEG) is responsible for supporting the company... ...research and development of large-scale video world models, including the design and construction... ...field.2. Solid foundation in deep learning algorithms and proven experience in large...Full timeRelocation package$213k - $263k
...company with the mission to be the world's most trusted driver. Since... ...the lifecycle of the machine learning workflow, including feature and experiment management, model development, optimization and... ...Simulation. We are looking for engineers with ML software or ML...Full timeRemote work- ...patients worldwide.We’re a team of engineers, clinicians, and innovators... ...outcomes around the world.The problems we solve demand... ...and fine-tune the foundation models—VFMs, VLMs, and VLA models—that... ...enable rapid prototyping and learning while building toward a product...Local areaWorldwideFlexible hours
- ...experiences. As a Senior Staff Engineer, you will lead the... ...techniques such as sequence modeling, deep learning, and large language models... ...Apply state-of-the-art machine learning and LLM techniques... ...engineering to solve complex real-world optimization problems. You...Hourly payWork at officeLocal areaRemote workFlexible hours
$193.3k - $261.5k
...AWS our vision is to make deep learning pervasive for everyday... ...the performance of complex ML models executed on AWS Inferentia and... ...role is for a senior software engineer in the Compiler team for AWS... ...AWSAmazon Web Services (AWS) is the world’s most comprehensive and...Local areaFlexible hours$195k - $230k
...About the RoleWe are looking for a Senior Machine Learning Engineer to help evolve our large-scale... ...and apply AI / LLM technologies to real-world production problems. You will work on... ...resolve issues related to data quality, model drift, and system performance in production...Full timeLocal areaWork from home$150k - $250k
...Plus was named by Fast Company as one of the World’s Most Innovative Companies. Partners... ...growing teams.We are seeking a highly skilled Machine Learning Engineer with deep expertise in developing Bird’s Eye View (BEV) fusion models using multimodal sensor inputs,...Full time$230k - $265k
...alongside industry-veteran scientists and engineers. As a Senior Machine Learning Engineer, you’ll bring your strong... ...for large language and speech models using PyTorch and/or JAX, making principled... ...operational excellence using real-world conversational data at scale.Set...Permanent employment$123.75k - $185k
...talent platform, trusted by the world’s largest & most respected... ..., and high standards. Our engineers, product leaders, and go-to-... ...and integrate large language models (LLMs) and other state-of-... ...Qualifications:Knowledge and passion in machine learning algorithms, Gen AI, LLMs,...Work experience placementWork at office3 days per week- ...measuring changes to the human and natural world will help them manage climate-... ...profile.Responsibilities:Develop Machine Learning (ML) and physics-based models for Earth systems applications... ...systems.Mandatory:Experience with earth engine and functional programming....Shift work
$128k - $260.5k
...One platform, its Zero Trust Engine, and the powerful NewEdge... ...Visit Careers at Netskope to learn more. Follow us on LinkedIn... ...intelligence (AI) and machine learning (ML) to protect the... ...Report Cards" for production models, tracking real-world accuracy, latency, and security...$193.3k - $261.5k
The Product: AWS Machine Learning accelerators are at the forefront of AWS... ...including silicon engineering, hardware design and verification... ...performance of complex neural net models on our custom-built AWS... ...of a compiler to handle the world's largest ML workloads. Architecting...InternshipLocal areaWork from homeRelocationFlexible hours- ...of patients worldwide.We’re a team of engineers, clinicians, and innovators united by one... ...faster, improving outcomes around the world.The problems we solve demand creativity... ...take a biopsy at a target location. As a machine learning engineer on the Ion project, you will...Local areaWorldwideFlexible hours
$224k - $356.5k
We are looking for outstanding Machine Learning Engineers to join our Physical AI teams. As the pioneers... ...AI that interacts with the physical world.This role is at the forefront of Physical... ...leverages state-of-the-art multimodal models and diffusion techniques to simulate...Full time- ...automate tasks. Powered by the world's most advanced LLMs, our proprietary models, and a sophisticated... ..., and continuously learn and adapt.Moveworks is trusted... ...Moveworks’ Reasoning Engine and natural language... ...software engineer with machine learning expertise to join...Work at officeImmediate startRemote workFlexible hours
$131k - $180k
...global leader in materials engineering solutions used to... ...advanced display in the world. We design, build and service... ...that encourages you to learn, develop, and grow your... ...cutting-edge AI and machine learning to accelerate... ...AI and large language models to tackle complex problems...Full time$150k - $230k
...About the RoleWe are looking for a hands-on Machine Learning Engineer to drive the post-training of our large language models, with a strong emphasis on reinforcement learning... ...product and business teams to translate real-world use cases into concrete training objectives...Full timeLocal areaWork from home- ...patients worldwide.We’re a team of engineers, clinicians, and innovators... ...outcomes around the world.The problems we solve demand... ...advanced medical imaging and machine learning algorithms tailored for advanced... ...(ViTs), DINO, Diffusion models, and Vision Language Models...Local areaWorldwideFlexible hours
$130k - $200k
Santa Clara, CAUS Engineering - Simulation /Full-time /HybridPlusAI... ...Fast Company as one of the World’s Most Innovative Companies.... ...teams.We are seeking a Senior Machine Learning Engineer with expertise in deep... ...scalable vehicle simulation models tailored to various vehicle...Full time- ...patients worldwide.We’re a team of engineers, clinicians, and innovators... ...outcomes around the world.The problems we solve demand... ...software and computer vision / machine learning algorithmsIterating with... ...up-to-date with the latest model architectures and development...Local areaWorldwideFlexible hours
$184k - $287.5k
We are seeking a Senior Machine Learning Engineer to join our end‑to‑end autonomous driving team! You... ...train, and deploy large‑scale E2E driving models that leverage VLM/VLA architectures,... ...improves our systems in the real world! Today, we’re tapping into the unlimited...Full time
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to Machine Learning Engineer - World Model. Be the first to apply!
- senior ml engineer Sunnyvale, CA
- machine learning engineer Sunnyvale, CA
- ai ml engineer Sunnyvale, CA
- machine learning software engineer Sunnyvale, CA
- computer vision machine learning engineer Sunnyvale, CA
- machine learning ai engineer Sunnyvale, CA
- machine learning research scientist Sunnyvale, CA
- data engineer machine learning Sunnyvale, CA
- internship machine learning Sunnyvale, CA
- machine learning scientist Sunnyvale, CA