Principal Generative-Ai Software Engineer

Details

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, and intelligent simulation. We welcome hard-working 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 novel world models (e.g., diffusion video models, VLMs, VLAs) for Physical AI 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 rapidly and iterate on experiments across cutting-edge AI domains, including text-to-image/video generation, reinforcement learning, reasoning, and foundation models.
• 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

• Stellar experience building and deploying generative AI systems (minimum 2 years industry or 3+ years research/postdoc).
• Proficiency in PyTorch, JAX, or other deep learning frameworks.
• Expertise in one or more of: diffusion models, auto-regressive models, VAE/GAN architectures, retrieval-augmented generation, neural rendering, or multi-agent systems.
• Deep familiarity with transformer architectures and variants of attention mechanisms.
• Hands-on experience with large scale training (ZeRO, DDP, FSDP, TP, CP) and data processing (Ray, Spark).
• Production-quality software engineering skills in Python.
• MS or PhD or equivalent experience in Computer Science, Machine Learning, Applied Math, Physics, or related field.
• 15+ years of relevant software development experience.

Ways to Stand Out

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

NVIDIA is widely considered one of the technology world's most desirable employers, leading in Artificial Intelligence, High-Performance Computing, and Visualization innovations. The GPU, their invention, serves as the visual cortex of modern computers and is central to their products and services.

Base salary range is 272,000 USD - 425,500 USD. Salary will be determined by location, experience, and pay of employees in similar positions. Eligible for equity and benefits. NVIDIA is an equal opportunity employer committed to diversity.