Principal Generative-AI Software Engineer
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 development and practical implementation, focusing on multimodal learning, video generation, and intelligent simulation.
The position requires a seasoned professional with 15+ years of software development experience and deep expertise in generative AI systems. You'll be working on designing and optimizing novel world models, including diffusion video models and vision-language models, while contributing to large-scale training infrastructure and high-efficiency inference pipelines.
NVIDIA, as the world leader in accelerated computing, offers an environment where you'll be working with some of the most forward-thinking professionals in the technology industry. The role combines hands-on technical work with leadership responsibilities, including mentoring junior engineers and contributing to open-source projects and technical papers.
Key technical requirements include proficiency in PyTorch or JAX, expertise in transformer architectures, and experience with large-scale training systems. The ideal candidate will have an advanced degree (MS/PhD) in Computer Science, Machine Learning, or a related field, with a strong background in production-quality software engineering.
The position offers a competitive salary range of $272,000 - $425,500 USD, along with equity and comprehensive benefits. Located in Santa Clara, CA, you'll be at the heart of NVIDIA's innovation center, working on projects that directly impact the future of AI computing.
This is an exceptional opportunity for someone who wants to push the boundaries of what's possible in AI while working with state-of-the-art technology and contributing to influential open-source projects. NVIDIA's commitment to diversity and inclusion, combined with their position as a technology leader, makes this an ideal role for someone looking to make a significant impact in the field of generative AI.
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
- 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, or related areas
- 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, or related field
- 15+ years of relevant software development experience
Benefits For Principal Generative-AI Software Engineer
- Equity and comprehensive benefits package
Jobs Related To NVIDIA Principal Generative-AI Software Engineer
