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:
Developer Technology Engineer - HPC and AI

Senior Developer Technology Engineer position at NVIDIA focusing on HPC and AI, requiring 3+ years experience and advanced degree, based in Seoul.

Senior Software Engineer, LLM Inference

Senior Software Engineer position at NVIDIA focusing on LLM Inference development, requiring expertise in C++, deep learning, and AI technologies.

Senior Perception Engineer

Senior Perception Engineer role at NVIDIA developing autonomous driving solutions using deep learning and computer vision, offering competitive salary and opportunity to work on cutting-edge technology.

Senior On-Device Model Inference Optimization Engineer

Senior AI optimization role at NVIDIA focusing on improving performance and efficiency of AI models for autonomous vehicles, offering competitive salary and equity benefits.

Senior Prediction and Planning Machine Learning Engineer - Autonomous Vehicles

Senior ML Engineer position at NVIDIA focusing on prediction and planning systems for autonomous vehicles

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