Taro Logo

Principal Generative-AI Software Engineer

NVIDIA is the world leader in accelerated computing, pioneering GPU technology and AI solutions.
$272,000 - $425,500
Machine Learning
Principal Software Engineer
In-Person
5,000+ Employees
15+ years of experience
AI

Description For Principal Generative-AI Software Engineer

NVIDIA is seeking a Principal Generative-AI Software Engineer to join their Cosmos generative AI engineering team. This role sits at the intersection of cutting-edge AI research and practical implementation, focusing on multimodal learning, video generation, synthetic data, and intelligent simulation. The position requires deep expertise in generative modeling and will involve working with state-of-the-art AI technologies.

The role combines hands-on technical work with leadership responsibilities, including mentoring junior engineers and contributing to both open-source and internal projects. You'll be working on novel world models, large-scale training infrastructure, and high-efficiency inference pipelines. The position requires extensive experience in deep learning frameworks, transformer architectures, and production-level software engineering.

NVIDIA offers a competitive compensation package with a base salary range of $272,000 - $425,500 USD, plus equity and comprehensive benefits. The company is known for being one of the technology world's most desirable employers, leading groundbreaking developments in AI, High-Performance Computing, and Visualization.

This is an excellent opportunity for someone with strong technical skills in AI/ML who wants to work at the forefront of generative AI technology. The role requires 15+ years of relevant software development experience and deep expertise in areas like diffusion models, auto-regressive models, and transformer architectures. NVIDIA's commitment to innovation and its position as a leader in accelerated computing makes this an exciting opportunity for experienced AI engineers looking to make a significant impact in the field.

Last updated 44 minutes ago

Responsibilities For Principal Generative-AI Software Engineer

  • 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
  • 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 Principal Generative-AI Software Engineer

Python
  • Minimum 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
  • Familiar 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, Applied Math, Physics
  • 15+ years of relevant software development experience

Benefits For Principal Generative-AI Software Engineer

Equity
  • Equity grants
  • Comprehensive benefits package

Interested in this job?

Jobs Related To NVIDIA Principal Generative-AI Software Engineer

Principal Deep Learning Software Engineer, LLM Performance

Principal Deep Learning Software Engineer role at NVIDIA focusing on LLM performance optimization and implementation using TensorRT, VLLM, and CUDA.

Principal Prediction and Planning Machine Learning Engineer - Autonomous Vehicles

Lead machine learning role focusing on prediction and planning systems for autonomous vehicles at NVIDIA.

Principal Machine Learning Engineer - Enterprise AI

Principal Machine Learning Engineer position at NVIDIA focusing on Enterprise AI solutions.

Principal Prediction and Planning Machine Learning Engineer - Autonomous Vehicles

Principal-level machine learning engineering role focused on prediction and planning systems for autonomous vehicles at NVIDIA.

Principal Deep Learning Software Engineer, LLM Performance

Principal Deep Learning Software Engineer position focused on LLM Performance optimization at NVIDIA's Santa Clara location.