Generative AI Engineering - Foundation Models

Employment Type: Full time

Position Overview

At NVIDIA, we're not just building the future, we're generating it. Our Cosmos generative AI engineering team is pushing the boundaries of what’s possible across multimodal learning, video generation, synthetic data, intelligent simulation, and agentic systems. We are looking for exceptionally driven engineers and applied scientists with deep experience in generative modeling to help define the next era of AI computing.

Responsibilities

• Design, post-train, and optimize foundation models (e.g., LLMs, diffusion video models, VLMs, VLAs) for real-world applications.
• Contribute to highly-collaborative development on large-scale training infrastructure, high-efficiency inference pipelines, and scalable data pipelines.
• Work with teams in research, software, and product to bring world models from idea to deployment.
• Collaborate on open-source and internal projects, author technical papers or patents, and mentor junior engineers.
• Prototype and iterate rapidly on experiments across cutting-edge AI domains, including agentic systems, reinforcement learning, reasoning, and video generation.
• Design and implement model distillation algorithms for size reduction and diffusion step optimization.
• Profile and benchmark training and inference pipelines to achieve production-ready performance requirements.

Requirements

• Experience: Minimum 8 years industry or 5+ years research/postdoc experience building and deploying generative AI systems.
• Deep Learning Frameworks: Proficiency in PyTorch, JAX, or other deep learning frameworks is a must.
• Generative Model Expertise: Expertise in one or more of the following: LLMs, coding agents, diffusion models, autoregressive models, VAE/GAN architectures, retrieval-augmented generation, neural rendering, or multi-agent systems.
• Transformer Architectures: Intimate familiarity with all variants of transformer architectures and attention mechanisms.
• Large-Scale Training & Data Processing: Hands-on experience with large-scale training (e.g., ZeRO, DDP, FSDP, TP, CP) and data processing (e.g., Ray, Spark).
• Software Engineering: Production-quality software engineering skills in Python.
• Education: MS or PhD or equivalent experience in Computer Science, Machine Learning, Applied Math, Physics, or a related field.
• Total Experience: 12+ years of relevant software development experience.

Ways to Stand Out

• Familiarity with high-performance computing and GPU acceleration.
• Contributions to influential open-source libraries or influential conference publications (NeurIPS, ICML, CVPR, ICLR).
• Experience working with multimodal data (e.g., vision-language, VLA, audio).
• Prior work with NVIDIA GPU-based compute clusters or simulation environments.

Company Information

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 on the planet working for us. If you're creative, passionate, and self-motivated, we want to hear from you! NVIDIA is leading the way in groundbreaking developments in Artificial Intelligence, High-Performance Computing and Visualization. The GPU, our invention, serves as the visual cortex of modern computers and is at the heart of our products and services.

Compensation:

• Base Salary Range:
  • Level 5: $224,000 - $356,500 USD
  • Level 6: $272,000 - $425,500 USD
• Eligibility: You will also be eligible for equity and benefits.

Application Deadline: Applications for this job will be accepted at least until July 29, 2025.

Equal Opportunity Employer: NVIDIA is committed to fostering a diverse 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, or disability.