Senior ML Engineer for Metropolis AI (Cosmos) – Equity

NVIDIA; Santa Clara (CA); On-site; USD 184,000 - 357,000; Full time

Benefits offered by this job

Equity

Benefits

Job summary

NVIDIA’s Metropolis team seeks a Senior ML Engineer to build production‑quality AI capabilities powered by Cosmos for intelligent video analytics and physical AI. You’ll collaborate with the Cosmos team to extend the platform and deliver real‑world perception solutions at scale.

Responsibilities include leading end‑to‑end ML projects, prioritizing features per the product roadmap, and working with cross‑functional teams to unblock execution. Equity and benefits are offered in this role.

Qualifications

• 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.

Responsibilities

• 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.

Skills

Deep learning

Diffusion models

LLMs

VLMs

Foundation models

Project leadership

End-to-end ML lifecycle

Education

MSc or PhD in CS/EE

Tools

CUDA

Triton

Job description

NVIDIA’s Metropolis team seeks a Senior ML Engineer to build production‑quality AI capabilities powered by Cosmos for intelligent video analytics and physical AI. You’ll collaborate with the Cosmos team to extend the platform and deliver real‑world perception solutions at scale.

Responsibilities include leading end‑to‑end ML projects, prioritizing features per the product roadmap, and working with cross‑functional teams to unblock execution. Equity and benefits are offered in this role.