Taro Logo

Senior Software Engineer, Generative-AI

NVIDIA is the world leader in accelerated computing, pioneering AI and digital twins technology.
$148,000 - $287,500
Machine Learning
Senior Software Engineer
In-Person
5,000+ Employees
5+ years of experience
AI

Description For Senior Software Engineer, Generative-AI

NVIDIA is seeking a Senior Software Engineer specializing in Generative AI to join their Cosmos team. This role sits at the cutting edge of AI development, focusing on multimodal learning, video generation, synthetic data, and intelligent simulation. The position offers a unique opportunity to work with state-of-the-art AI technologies and contribute to groundbreaking developments in the field.

The role combines deep technical expertise in generative AI with practical software engineering skills. You'll be working on novel world models, large-scale training infrastructure, and high-efficiency inference pipelines. The position requires significant experience in generative AI systems, with expertise in frameworks like PyTorch and JAX, and deep understanding of transformer architectures.

NVIDIA offers a competitive compensation package, with base salary ranging from $148,000 to $287,500 depending on level and experience. The company is known for its innovative culture and is consistently ranked as one of the technology world's most desirable employers. You'll be working with some of the industry's brightest minds on projects that directly impact the future of AI computing.

The ideal candidate will have 5+ years of software development experience, with a strong background in machine learning and generative AI. Additional expertise in high-performance computing, GPU acceleration, or contributions to influential open-source libraries would be highly valued. This is an excellent opportunity for someone passionate about pushing the boundaries of AI technology while working on practical, production-ready solutions.

Last updated 6 days ago

Responsibilities For Senior Software Engineer, Generative-AI

  • Design, post-train, and optimize novel world models for Physical AI applications
  • Contribute to development on large-scale training infrastructure and inference pipelines
  • Work with teams 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 on experiments across AI domains
  • Design and implement model distillation algorithms
  • Profile and benchmark training and inference pipelines

Requirements For Senior Software Engineer, Generative-AI

Python
  • 2+ years industry or 3+ years research/postdoc in generative AI systems
  • Proficiency in PyTorch, JAX, or other deep learning frameworks
  • Expertise in diffusion models, auto-regressive models, VAE/GAN architectures, or related areas
  • Familiarity with transformer architectures and attention mechanisms
  • Experience with large scale training and data processing
  • Production-quality software engineering skills
  • MS or PhD or equivalent experience in Computer Science, Machine Learning, or related field
  • 5+ years of relevant software development experience

Benefits For Senior Software Engineer, Generative-AI

Equity
  • Equity

Jobs Related To NVIDIA Senior Software Engineer, Generative-AI