Senior ML Software 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
• Proficiency in PyTorch, JAX, or other deep learning frameworks
• Expertise in generative AI systems
• Experience with large scale training and data processing
• MS or PhD in Computer Science, Machine Learning, Applied Math, Physics, or related field

Responsibilities

• Design, post-train, and optimize foundation models for real world applications
• Contribute to development on large-scale training infrastructure and data pipelines
• Work with teams to bring world models from idea to deployment
• Collaborate on open-source and internal projects
• Prototype and iterate rapidly on experiments across cutting-edge AI domains
• Design and implement model distillation algorithms
• Profile and benchmark training and inference pipelines

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