Taro Logo

Senior Deep Learning Software Engineer, PyTorch

NVIDIA is the world leader in accelerated computing, pioneering solutions in AI and digital twins.
$184,000 - $356,500
Machine Learning
Senior Software Engineer
In-Person
5+ years of experience
AI
This job posting may no longer be active. You may be interested in these related jobs instead:

Description For Senior Deep Learning Software Engineer, PyTorch

NVIDIA is seeking a Senior Deep Learning Software Engineer to join their PyTorch team. This role offers an opportunity to work at the forefront of AI technology, designing and building tools used by AI engineers worldwide. The position involves working with PyTorch, a leading deep learning framework, and collaborating with teams across NVIDIA and the international PyTorch community. The ideal candidate will have strong expertise in high-performance computing, Python/C++ programming, and distributed systems. They will be responsible for developing PyTorch components that scale across thousands of GPUs, improving network and GPU efficiency, and supporting production AI solutions. The role offers competitive compensation, including a base salary range of $184,000 - $356,500, equity, and comprehensive benefits. NVIDIA is known for its innovative culture and is consistently ranked among the technology world's most desirable employers. The company is committed to diversity and inclusion, fostering an environment where creative and autonomous engineers can thrive while working on cutting-edge technology that impacts various industries globally.

Last updated 6 months ago

Responsibilities For Senior Deep Learning Software Engineer, PyTorch

  • Design and build PyTorch components and tools that run efficiently on Super Computers with 1000s of GPUs
  • Collaborate with NVIDIA's hardware and software teams to improve the network and GPU efficiency in PyTorch
  • Design, build and support production AI solutions used by enterprise customers and partners
  • Work with internal applied researchers to improve their AI tools

Requirements For Senior Deep Learning Software Engineer, PyTorch

Python
  • MS in Computer Science or Engineering (or equivalent experience) with 5+ years professional experience in High Performance Computing
  • Proficient with Python C++ programming
  • Proven experience with Thread and Distributed Parallel Programming
  • Demonstrated experience developing large software projects
  • Strong verbal and written communication skills

Benefits For Senior Deep Learning Software Engineer, PyTorch

Medical Insurance
Dental Insurance
Vision Insurance
Equity
  • Competitive salaries
  • Equity
  • Comprehensive benefits package

Interested in this job?