Senior Software Engineer, Generative-AI

16 Days Old

Join to apply for the Senior Software Engineer, Generative-AI role at NVIDIA Join to apply for the Senior Software Engineer, Generative-AI role at NVIDIA Get AI-powered advice on this job and more exclusive features. 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.
What You'll Be Doing
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.
What We Need To See
We are looking for 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 is a must! We are working on all range of foundation models. You should have expertise in one or more of: diffusion models, auto-regressive models, VAE/GAN architectures, retrieval-augmented generation, neural rendering, or multi-agent systems. Our models are predominantly built on the transformer architectures. You should be intimately familiar with all variants of the attention mechanisms. Hands on experience with large scale training (e.g., ZeRO, DDP, FSDP, TP, CP) and data processing (e.g. Ray, Spark). All we do is in Python and we open source our product, therefore production-quality software engineering skills is highly relevant. MS or PhD or equivalent experience in Computer Science, Machine Learning, Applied Math, Physics, or a related field. 5+ years of relevant software development experience
Ways To Stand Out From The Crowd
Familiarity with high-performance computing and GPU acceleration. Contributions to influential open-source libraries or influential conference publications (NeurIPS, ICML, CVPR, ICLR). Experience working with multimodal data (e.g., vision-language, VLA, audio). Prior work with NVIDIA GPU-based compute clusters or simulation environments.
NVIDIA is widely considered to be one of the technology world’s most desirable employers. We have some of the most forward-thinking and hardworking people on the planet working for us. If you're creative, passionate and self-motivated, we want to hear from you! NVIDIA is leading the way in groundbreaking developments in Artificial Intelligence, High-Performance Computing and Visualization. The GPU, our invention, serves as the visual cortex of modern computers and is at the heart of our products and services.
The base salary range is 148,000 USD - 287,500 USD. Your base salary will be determined based on your location, experience, and the pay of employees in similar positions.
You will also be eligible for equity and benefits . NVIDIA accepts applications on an ongoing basis.
NVIDIA is committed to fostering a diverse work environment and proud to be an equal opportunity employer. As we highly value diversity in our current and future employees, we do not discriminate (including in our hiring and promotion practices) on the basis of race, religion, color, national origin, gender, gender expression, sexual orientation, age, marital status, veteran status, disability status or any other characteristic protected by law.
JR1997674
Seniority level Seniority level Mid-Senior level Employment type Employment type Full-time Job function Industries Computer Hardware Manufacturing, Software Development, and Computers and Electronics Manufacturing Referrals increase your chances of interviewing at NVIDIA by 2x Sign in to set job alerts for “Senior Software Engineer” roles. San Mateo, CA $90,000.00-$150,000.00 12 months ago Senior Backend Software Engineer, TikTok Customer Service Platform San Jose, CA $194,000.00-$355,000.00 1 week ago Sr. Software Engineer, Supply Chain Applications Mountain View, CA $204,000.00-$259,000.00 8 hours ago Senior Software Development Engineer, Virtual Network San Jose, CA $194,000.00-$410,000.00 4 weeks ago Tech Lead Software Engineer, TikTok Customer Service Platform San Jose, CA $224,000.00-$410,000.00 1 week ago Sr Principal Engineer Software (AIOps for NGFW) Senior Frontend Software Engineer, TikTok Customer Service Platform San Jose, CA $194,000.00-$355,000.00 1 week ago Senior Developer Events Manager, Google Cloud Sunnyvale, CA $165,000.00-$245,000.00 2 days ago Senior Software Engineer, Fabric Networking - GPU Santa Clara, CA $148,000.00-$287,500.00 3 days ago Mountain View, CA $204,000.00-$259,000.00 1 week ago Senior Software Engineer, Networking Software Santa Clara, CA $148,000.00-$287,500.00 3 days ago Senior Software Engineer, Metrics and Evaluation - Autonomous Vehicles Senior Software Engineer - Localization and Mapping (SLAM) San Francisco Bay Area $180,000.00-$260,000.00 4 weeks ago San Francisco Bay Area $160,000.00-$300,000.00 3 weeks ago Sr. Software Engineer, Plant Modeling and Tools Software Engineering Technical Leader - Silicon One Customer Engineering San Jose, CA $173,100.00-$241,700.00 3 days ago Senior/Staff Software Engineer, Mission Planning Foster City, CA $217,000.00-$332,000.00 1 week ago Senior System Software Engineer, Power and Thermal Management Sr Principal Engineer Software (AIOps for NGFW) Redwood City, CA $79,800.00-$178,100.00 1 month ago San Mateo, CA $448,990.00-$515,210.00 1 week ago Sunnyvale, CA $175,000.00-$245,000.00 4 months ago Senior Software Engineer, ASIC Verification Tools Software Engineering Technical Leader (Hybrid Remote- San Jose, CA) San Jose, CA $198,600.00-$282,900.00 5 days ago Palo Alto, CA $135,100.00-$168,900.00 2 weeks ago Senior Staff Software Engineer, Managed AI We’re unlocking community knowledge in a new way. Experts add insights directly into each article, started with the help of AI.
#J-18808-Ljbffr
Location:
Santa Clara, CA, United States
Job Type:
FullTime
Category:
IT & Technology