NU
Full time

Senior Machine Learning Engineer, Cosmos Engineering

NVIDIA USA
US, CA, Santa Clara 2 days ago
AI overview

This role involves building and delivering AI solutions for intelligent video analytics and perception using foundation models. The engineer will identify areas for model improvement, partner with platform teams on feature development, and lead open-sourcing efforts. A PhD or MSc in a related field with 8+ years of applied ML experience, including expertise in generative architectures and large-scale models, is required.

NVIDIA's Metropolis team advances physical AI by building intelligent video analytics and perception solutions for smart cities, industrial automation, and autonomous systems at scale. We seek a Senior ML Engineer to compose and deliver next-generation Metropolis solutions powered by Cosmos, NVIDIA's world foundation model platform. Your main task will be constructing production-quality AI capabilities for Metropolis by demonstrating Cosmos to address real-world perception and physical AI challenges. You will also partner closely with the Cosmos team to develop and broaden the platform’s features aligned with the Metropolis product roadmap. If you are driven to push generative AI, simulation, and large-scale model development forward and want your work to impact real-world systems, we invite you to apply.

What you'll be doing:

  • Build and deliver innovative Metropolis AI solutions powered by Cosmos world foundation models, addressing real-world requirements in intelligent video analytics, perception, and physical AI.

  • Find areas where Cosmos models underperform or lack capability and recommend new solutions. These involve generating synthetic data, refining tuning methods, and improving architecture. Metropolis use cases serve as the main reference.

  • Partner with the Cosmos team to identify and prioritize new platform features guided by the Metropolis product roadmap, contributing to deliverables that support the wider ecosystem.

  • Lead the open-sourcing of solutions and research artifacts developed by the team, contributing to the broader AI and research community.

  • Stay current with the latest advances in foundation models, generative architectures, and training methodologies, and actively bring relevant insights back to the team.

  • Partner multi-functionally with Product, Program, Engineering, and Data Procurement teams to drive alignment and unblock execution.

What we need to see:

  • MSc or PhD in Computer Science, Electrical Engineering, or a related field — or equivalent experience.

  • 8+ years of proven experience in applied machine learning or AI research.

  • Deep expertise in deep learning fundamentals, with hands-on experience in diffusion models and generative architectures.

  • Experience in pre-training or refining large language models (LLMs), vision-language models (VLMs), or world foundation models (WFMs).

  • Experience working with large-scale foundation models, including training workflows, fine-tuning techniques, and evaluation approaches.

  • Experience working with simulation environments like Isaac Sim or similar platforms.

  • Consistent track record of leading projects end-to-end and delivering results with clarity and accountability.

  • Ability to manage and complete tasks across multiple parallel workstreams in a fast-paced, evolving environment.

  • End-to-end understanding of ML development and deployment life cycle with the ability to quickly adopt and bring to bear modern AI development tools and workflows.

Ways to stand out from the crowd:

  • Hands-on experience with model compression, quantization, and real-time inference optimization for production deployments.

  • Previous experience implementing AI solutions in physical settings such as public areas, smart infrastructure, or robotics platforms.

  • Experience scaling AI systems across distributed infrastructure, including multi-node training and large-scale data pipelines.

  • Published research or open-source contributions in relevant areas such as generative models, synthetic data, or physical AI.

  • Familiarity with CUDA, Triton, or low-level GPU kernel development for inference pipeline acceleration.

Your base salary will be determined based on your location, experience, and the pay of employees in similar positions. The base salary range is 184,000 USD - 287,500 USD for Level 4, and 224,000 USD - 356,500 USD for Level 5.

You will also be eligible for equity and benefits.

Applications for this job will be accepted at least until November 2, 2026.

This posting is for an existing vacancy.

NVIDIA uses AI tools in its recruiting processes.

Key details

ContractFull time
Posted2 days ago
Senior Revenue Data and Information System Support Analyst
NVIDIA USA · US, CA, Santa Clara
Senior Revenue and Pricing Business Analyst
NVIDIA USA · US, CA, Santa Clara
Compiler Software Engineering Intern - Summer 2027
NVIDIA USA · US, WA, Redmond
AI Systems Software Engineering Intern – Summer 2027
NVIDIA USA · US, TX, Austin
VLSI CAD Timing Intern, Applied AI - Summer 2027
NVIDIA USA · US, CA, Santa Clara
Senior HPC Storage Engineer - EDA Infrastructure
NVIDIA USA · US, CA, Santa Clara | US, Remote
Senior Infrastructure Security Engineer – EDA Clusters
NVIDIA USA · US, CA, Remote | US, NC, Remote | US, CO, Remote | US, WA, Remote | US, MA, Remote
ASIC Physical Design Engineer, Netlisting - New College Grad 2026
NVIDIA USA · US, CA, Santa Clara | US, TX, Austin
Optical Package Test Engineer
NVIDIA USA · US, CA, Santa Clara
Senior Manager, Engineering - Data Center Firmware
NVIDIA USA · US, CA, Santa Clara | US, CA, Remote
Senior Staff Software Engineer - Agentic IT Operations
NVIDIA USA · US, CA, Santa Clara
Senior Analytics Engineer, Finance
NVIDIA USA · US, CA, Santa Clara
Senior Software Engineer, AI Frameworks
NVIDIA USA · US, CA, Santa Clara | US, CA, Remote
Senior Software Engineer - Simulation
NVIDIA USA · US, CA, Santa Clara
Senior High-Performance LLM Training Engineer
NVIDIA USA · US, CA, Santa Clara
Senior HPC Architect, Automation and At-Scale Deployment
NVIDIA USA · US, CA, Santa Clara | US, TX, Remote | US, CO, Remote | US, WA, Remote | US, CA, Remote | US, Remote | US, AR, Remote | US, NC, Remote | US, IL, Remote | US, OR, Remote | US, UT, Remote | US, NM, Remote | US, MA, Remote
ML Ops DSX Senior Engineer
NVIDIA USA · US, CA, Santa Clara | US, Remote | US, WA, Seattle
Silicon Product Engineer – Manufacturing
NVIDIA USA · US, CA, Santa Clara
Production Application Engineer, Enterprise
NVIDIA USA · Taiwan, Taipei | Taiwan, Hsinchu
SAP SD Business Systems Analyst
NVIDIA USA · India, Pune | India, Hyderabad

Frequently asked questions