Senior ML Engineer, Metropolis

As a Senior ML Engineer at NVIDIA, you will be part of a team harnessing the boundless capabilities of AI to build the next era of computing.

Your responsibilities will include:

• Finding areas where Cosmos models underperform or lack capability and recommending new solutions
• Partnering with the Cosmos team to identify and prioritize new platform features
• Leading the open-sourcing of solutions and research artifacts developed by the team
• Staying current with the latest advances in foundation models, generative architectures, and training methodologies
• Partnering multi-functionally with various teams to drive alignment and unblock execution

To be successful in this role, you will need:

• MSc or PhD in Computer Science, Electrical Engineering, or a related field
• 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, vision-language models, or world foundation models
• 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

Benefits include a competitive salary package, equity, and benefits.