Taro Logo

Senior Deep Learning Software Engineer, Recipe Pathfinding

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

Description For Senior Deep Learning Software Engineer, Recipe Pathfinding

NVIDIA is seeking a Senior Deep Learning Software Engineer for Recipe Pathfinding to revolutionize SW systems for discovering low-precision and sparsity recipes in LLMs. This role focuses on creating infrastructure and tooling to accelerate the discovery of efficiency-gaining transformations in large language models. The position involves working with cutting-edge hardware features on Blackwell, Rubin, and future platforms, spanning all phases of the LLM lifecycle.

The role is heavily focused on coding, infrastructure development, and performance optimization, with direct impact on NVIDIA's internal software systems and products like Megatron-LM and Transformer Engine. The work is critical for minimizing computational costs in production runs that can reach millions of dollars.

As a world leader in accelerated computing, NVIDIA offers an exciting opportunity to work with some of the most forward-thinking professionals in the technology industry. The company has transformed from its origins in computer graphics to become "the AI computing company," with GPUs now powering deep learning algorithms and acting as the brain of computers, robots, and self-driving cars.

The position offers competitive compensation, including a base salary range of $184,000 - $356,500 USD, equity, and comprehensive benefits. The successful candidate will join a team committed to developing next-generation software solutions and will have the opportunity to influence the future of AI computing platforms.

Last updated 2 minutes ago

Responsibilities For Senior Deep Learning Software Engineer, Recipe Pathfinding

  • Create well-designed and well-tested SW systems and PoCs to support recipe exploration for research settings
  • Analyze and prototype state-of-the-art methods for quantization and sparsity
  • Benchmark, profile, and optimize LLM workloads in cluster settings
  • Keep developer and researcher productivity and efficiency high by removing obstacles
  • Participate in code reviews, address code review feedback if applicable

Requirements For Senior Deep Learning Software Engineer, Recipe Pathfinding

Python
  • PhD, M.S. degree or equivalent experience in Computer Science or related field
  • 5+ years of experience in a relevant software engineering role
  • Strong software engineering background with focus on building concise and well-tested code
  • Proficient in Python
  • Firsthand experience working with ML accelerators
  • Experience with PyTorch or similar framework
  • Experience with performance optimization and debugging
  • Strong written and oral communication skills

Benefits For Senior Deep Learning Software Engineer, Recipe Pathfinding

Equity
  • Equity

Interested in this job?

Jobs Related To NVIDIA Senior Deep Learning Software Engineer, Recipe Pathfinding

Full Stack Developer, AI and LLM

Full Stack Developer position focusing on AI/LLM integration for NVIDIA's Silicon Solutions Group, developing automation infrastructure and AI solutions.

Full Stack Developer, AI and LLM

Senior Full Stack Developer position at NVIDIA focusing on AI and LLM development, offering competitive salary and hybrid work arrangement.

Senior Deep Learning Software Engineer, Recipe Pathfinding

Senior Deep Learning Software Engineer position at NVIDIA focusing on recipe pathfinding and AI development.

Senior Deep Learning Software Engineer, Recipe Pathfinding

Senior Deep Learning Software Engineer position at NVIDIA focusing on recipe pathfinding and AI algorithm development.

Senior Performance Software Engineer, Deep Learning Libraries

Senior Performance Software Engineer position focused on developing and optimizing deep learning libraries at NVIDIA.