Senior Deep Learning Software Engineer, Inference and Model Optimization

Nvidia

Confirmed live today High trust
Remote

Quick summary

Work type
Remote
Location
Santa Clara, CA
Salary
$184,000–$287,500 / yr
Employment
Full-time
Posted
18 days ago
Freshness
Confirmed live today

Market check

Salary context

Competitive pay

How this pay compares to similar roles

Similar $223k
This role $236k
$162k $301k
below market most similar roles pay here above market

This role pays more than 61% of similar roles. Most pay $191,875–$254,750 — the blue band above. At the midpoint, this role pays about $236k versus about $223k 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 1623 open roles on FindRole.

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

Most-posted roles

View all roles at Nvidia

At a glance

TL;DR · Senior Deep Learning Software Engineer, Inference and Model Optimization

The Senior Deep Learning Software Engineer, Inference and Model Optimization joins the Algorithmic Model Optimization Team to develop and scale automated inference and deployment solutions. This role involves training, developing, and deploying generative AI models like large language models and diffusion models. You will analyze and extract standardized model graph representations from arbitrary models to build a modular software platform. Key responsibilities include developing high-performance optimization techniques such as automated model sharding, tensor parallelism, and efficient attention kernels with kv-caching. You will utilize the PyTorch 2.0 ecosystem, HuggingFace, CUDA, TRT-LLM, and Triton to improve inference efficiency. The work focuses on solving hardware and software optimization opportunities by profiling GPU kernel-level performance to maintain leadership in inference software solutions like TensorRT and TRT Model Optimizer.

What you'll do

  • Train, develop, and deploy generative AI models like LLMs and diffusion models using NVIDIA's AI software stack.
  • Extract standardized model graph representations from arbitrary PyTorch models using the torch 2.0 ecosystem.
  • Develop high-performance optimization techniques including automated model sharding, tensor parallelism, and efficient attention kernels.
  • Analyze and profile GPU kernel-level performance to identify hardware and software optimization opportunities.
  • Architect and design a modular, scalable software platform to support broad model optimization and user experience.
  • Implement high-performance kernels using CUDA, TRT-LLM, and Triton for automated deployment solutions.
  • Innovate inference performance to maintain market leadership for TRT, TRT-LLM, and TRT Model Optimizer.

What we're looking for

  • Master's, PhD, or equivalent experience in Computer Science, AI, Applied Math, or a related field.
  • 8+ years of relevant work or research experience in Deep Learning.
  • Excellent software design skills, including debugging, performance analysis, and test design.
  • Strong proficiency in Python, PyTorch, and related ML tools like HuggingFace.
  • Strong algorithms and programming fundamentals.
  • Good written and verbal communication skills with the ability to work independently and collaboratively.
  • Contributions to PyTorch, JAX, or other Machine Learning Frameworks (preferred).
  • Knowledge of GPU architecture, compilation stack, and experience writing high-performance GPU kernels in CUDA, CUTLASS, or Triton (preferred).

More like this

Similar roles

Senior Software Engineer, AI Inference Performance

Nvidia

Santa Clara, CA 29 days ago $184,000–$287,500
CUDA Python C++ Rust TensorRT-LLM vLLM SGLang Triton PyTorch NVIDIA Nsight Systems CUTLASS NCCL Quantization Distributed Systems GPU Architecture Model Parallelism Speculative Decoding
6+ yrs exp

Senior Deep Learning Software Engineer, Inference

Nvidia

Remote (Santa Clara, CA) 47 days ago $184,000–$287,500
CUDA C++ Python vLLM SGLang PyTorch Triton CUTLASS NCCL NVSHMEM FlashInfer Deep Learning GPU Programming High-Performance Computing Model Serving Performance Optimization Agile
6+ yrs exp Remote