Senior ML Engineer, Metropolis

Details

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. The Senior ML Engineer will develop next-generation Metropolis solutions powered by Cosmos, NVIDIA's world foundation model platform, and partner with the Cosmos team to expand platform capabilities aligned with the Metropolis product roadmap.

Responsibilities

• Build and deliver production-quality Metropolis AI solutions powered by Cosmos world foundation models for intelligent video analytics, perception, and physical AI use cases.
• Identify areas where Cosmos models underperform or lack capabilities, and recommend solutions involving synthetic data generation, improved tuning methods, and architectural enhancements.
• Partner with the Cosmos team to identify and prioritize platform features based on the Metropolis product roadmap.
• Lead the open-sourcing of solutions and research artifacts developed by the team.
• Stay current with advances in foundation models, generative architectures, and training methodologies, and apply relevant insights to team projects.
• Collaborate with Product, Program, Engineering, and Data Procurement teams to drive alignment and unblock execution.

Requirements

• MSc or PhD in Computer Science, Electrical Engineering, or a related field, or equivalent experience.
• 8+ years of 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 pre-training or refining large language models (LLMs), vision-language models (VLMs), or world foundation models (WFMs).
• Experience with large-scale foundation model training workflows, fine-tuning techniques, and evaluation approaches.
• Experience with simulation environments such as Isaac Sim or similar platforms.
• A track record of leading projects end-to-end and delivering results with clarity and accountability.
• Ability to manage multiple parallel workstreams in a fast-paced, evolving environment.
• End-to-end understanding of the machine learning development and deployment lifecycle, with the ability to adopt modern AI development tools and workflows.

Preferred Qualifications

• Experience with model compression, quantization, and real-time inference optimization for production deployments.
• 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 generative models, synthetic data, or physical AI.
• Familiarity with CUDA, Triton, or low-level GPU kernel development for inference pipeline acceleration.

Compensation and Benefits

• Base salary range: $184,000–$287,500 for Level 4.
• Base salary range: $224,000–$356,500 for Level 5.
• Eligibility for equity and benefits.
• Applications will be accepted at least until August 21, 2026.
• NVIDIA is an equal opportunity employer and is committed to an inclusive work environment.