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

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 3 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?

Jobs Related To NVIDIA Senior Deep Learning Software Engineer, PyTorch

Senior AI and LLM Solutions Software Engineer

Senior AI Engineer role at NVIDIA focusing on developing AI and LLM solutions for chip design and verification processes, requiring 5+ years of software development experience.

Senior ASIC Infrastructure Engineer

Senior ASIC Infrastructure Engineer role at NVIDIA focusing on AI/ML applications for chip design, offering competitive compensation and opportunity to work with cutting-edge technology.

Technical Marketing Engineer - AI Platform Software

Senior Technical Marketing Engineer role at NVIDIA focusing on AI Platform Software, requiring 5+ years experience and deep learning expertise.

AI Algorithm Engineer - Silicon Solution Group

Senior AI Algorithm Engineer role at NVIDIA's Silicon Solution Group, focusing on AI model implementation and optimization for chip solutions.

Senior Deep Learning Performance Architect

Senior Deep Learning Performance Architect role at NVIDIA focusing on developing high-performance, energy-efficient architectures for AI and machine learning applications.