Senior Machine Learning Engineer, Cosmos Engineering

NVIDIA Corporation; Santa Clara (CA); Presencial; USD 184.000 - 357.000; Jornada completa

Ventajas ofrecidas por este puesto de trabajo

Equity

Descripción de la vacante

NVIDIA Corporation in Santa Clara, CA, is seeking a senior ML researcher to build Metropolis AI solutions atop Cosmos foundation models for intelligent video analytics and perception.

You will lead end-to-end projects, collaborate with product, engineering, and data procurement teams, and contribute to open-source artifacts. Requires 8+ years in ML, MSc/PhD, diffusion models, LLMs/VLMs, and large-scale foundation models.

Formación

• MSc/PhD in CS, EE or 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 diffusion models and generative architectures.
• Experience in pre-training or refining LLMs, VLMs, or WFMs.
• Experience with large-scale foundation models, training workflows, fine-tuning techniques, and evaluation approaches.

Responsabilidades

• Build and deliver Metropolis AI solutions powered by Cosmos models for intelligent video analytics and perception.
• Identify model underperformance and propose improvements including synthetic data generation and architecture refinements.
• Collaborate with Cosmos team to shape platform features aligned with product roadmap.

Conocimientos

Diffusion models

Generative architectures

LLMs

VLMs

World foundation models

End-to-end project leadership

Educación

MSc or PhD in Computer Science, Electrical Engineering, or related field

Herramientas

CUDA

Triton

Simulation environments (e.g., Isaac Sim)

Descripción del empleo

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.

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.

With a competitive salary package and benefits, NVIDIA is widely considered to be one of the technology world’s most desirable employers. We have some of the most forward-thinking and hardworking people in the world working for us.

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 October 11, 2026.

NVIDIA uses AI tools in its recruiting processes.

NVIDIA is committed to fostering an inclusive work environment and proud to be an equal opportunity employer. As we highly value diversity in our current and future employees, we do not discriminate (including in our hiring and promotion practices) on the basis of race, religion, color, national origin, gender, gender expression, sexual orientation, age, marital status, veteran status, disability status or any other characteristic protected by law.