Senior Generative AI Research Engineer

Job Description

NVIDIA is seeking driven engineers and applied scientists with deep experience in generative modeling to help define the next era of AI computing.

Requirements

• Minimum 8 years industry or 5+ years research/postdoc experience in building and deploying generative AI systems
• Proficiency in PyTorch, JAX, or other deep learning frameworks
• Expertise in one or more of: LLMs, coding agents, diffusion models, autoregressive models, VAE/GAN architectures, retrieval-augmented generation, neural rendering, or multi-agent systems
• Hands-on experience with large scale training and data processing
• Production-quality software engineering skills
• MS or PhD in Computer Science, Machine Learning, Applied Math, Physics, or related field
• 12+ years of relevant software development experience

Responsibilities

• Design, post-train, and optimize foundation models for real-world applications
• Contribute to collaborative development on training infrastructure, inference pipelines, and data pipelines
• Work with research, software, and product teams to bring 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
• Design and implement model distillation algorithms for size reduction and diffusion step optimization
• Profile and benchmark training and inference pipelines for production-ready performance

Benefits

• Multiple relocation packages
• Two weeklong shutdowns (mid-summer and year-end) in the US (in addition to PTO)
• 8-week parental leave
• 9 Employee Resource Groups
• Annual bonus offering
• Flexible work arrangements
• Up to 6% 401K matching