Senior Software Engineer, PyTorch, Deep Learning

Nvidia

Remote

Quick summary

Work type
Remote
Location
Remote
Salary
$152,000–$241,500 / yr
Posted
6 days ago

Market check

Salary context

Competitive pay

How this pay compares to similar roles

Similar $220k
This role $197k
$139k most similar roles pay here $274k

This role pays less than 65% of similar roles. Most pay $193,000–$246,150 — the shaded band above. At the midpoint, this role pays about $197k versus about $220k for comparable roles.

Based on 240 similar postings.

Employer

About Nvidia

Nvidia is a leading designer of graphics processing units (GPUs) and system-on-chip units, powering gaming, professional visualization, data centers, and artificial intelligence workloads. Industry: Semiconductors & AI Computing

Nvidia currently has 942 open roles on FindRole.

Listed pay typically runs $184,000–$287,500 across 931 roles with salary data.

Most-posted roles

View all roles at Nvidia

At a glance

TL;DR · Senior Software Engineer, PyTorch, Deep Learning

Role Posting Log in to save

As a Senior Deep Learning Software Engineer specializing in PyTorch at NVIDIA, you will join an ambitious team contributing to the company’s AI platform and directly enhancing PyTorch. Your daily tasks include designing efficient PyTorch components for supercomputers with thousands of GPUs, collaborating with hardware and software teams to optimize GPU performance, supporting enterprise customers with production AI solutions, and working alongside internal researchers to improve their tools. The ideal candidate has a strong background in C++ programming, systems software, and distributed parallel computing, along with experience in large-scale project development and multi-disciplinary teamwork. Familiarity with CUDA, Python, deep learning compilers, and modeling trends is beneficial, as is active participation in the open-source community. This role offers an unparalleled opportunity to influence AI development on a global scale within one of the industry’s leading technology companies.

Skills

PyTorch C++ CUDA Python Distributed Parallel Programming Thread Deep Learning Compilers Deep Learning Modeling Trends Multi-disciplinary Teams CI/CD

What you'll do

  • Design PyTorch components to run efficiently on supercomputers with thousands of GPUs.
  • Collaborate with NVIDIA teams to enhance GPU performance in PyTorch.
  • Build and support production AI solutions for enterprise customers and partners.
  • Work with internal researchers to improve their AI tools and technologies.
  • Contribute to the development and improvement of large-scale software projects.

What we're looking for

  • 3+ years of professional experience in deep learning.
  • Proficient in C++ programming and distributed parallel programming.
  • Strong understanding of systems software and interfaces.
  • Experience developing large-scale software projects.
  • Contributions to the open source community, particularly PyTorch.
  • Familiarity with CUDA Programming and Python.
  • Excellent verbal and written communication skills.

Related searches

More Senior Software Engineer jobs Remote Senior Software Engineer jobs All jobs at Nvidia

More like this

Similar roles

Senior Deep Learning Software Engineer

Nvidia

Santa Clara, CA +1 65 days ago $224,000–$356,500
Python PyTorch JAX CUDA TensorRT NVIDIA_Triton CUTLASS GPU_optimization Deep_Learning Inference_Optimization Speculative_decoding LoRA Performance_profiling Algorithm_design Machine_Learning_Frameworks
Hybrid
Save

Senior Deep Learning Software Engineer, TensorRT Performance

Nvidia

Remote (Santa Clara, CA) 98 days ago $152,000–$241,500
C++ Python TensorRT PyTorch JAX TensorFlow ONNX CUDA GPU Transformers Recommenders ASR TTS Visual_Understanding TorchDynamo TorchInductor vLLM Jetson_Systems Low_Latency_Optimization
Remote
Save

Senior Software Engineer, Deep Learning

Nvidia

Santa Clara, CA 170 days ago $184,000–$287,500
PyTorch JAX TensorFlow ONNX MLIR TVM Triton CUDA C++ C C++ LLVM Deep Learning GPU Programming Compiler Technologies Low Precision Inference Quantization Compression of DNNs Graph Compiler Kernel Generator
Save