Senior Machine Learning Engineer, Cosmos Engineering

About the role

For more than 25 years, NVIDIA has been driving innovation in computer graphics, PC gaming, and accelerated computing. It’s a distinctive heritage of creativity driven by excellent technology—and outstanding people. Today, we’re harnessing the boundless capabilities of AI to build the next era of computing. An era where our GPU serves as the intelligence behind computers, robots, and autonomous vehicles that perceive the world. Doing what’s never been done before takes vision, innovation, and the world’s best talent. As an NVIDIAN, you’ll be part of a varied, encouraging environment where everyone is motivated to perform at their highest level. Join our team and discover how you can build a lasting impact on the world.

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.