Senior Software Engineer, Deep Learning - MLIR TRT

Nvidia

Quick summary

Work type
On-site
Location
Santa Clara, CA
Salary
$184,000–$287,500 / yr
Posted
153 days ago
Nearby
99+ roles within 25 mi

Market check

Salary context

Above market

How this pay compares to similar roles

Similar $215k
This role $236k
$158k most similar roles pay here $301k

This role pays more than 72% of similar roles. Most pay $191,656–$239,212 — the shaded band above. At the midpoint, this role pays about $236k versus about $215k 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 987 open roles on FindRole.

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

Most-posted roles

View all roles at Nvidia

At a glance

TL;DR · Senior Software Engineer, Deep Learning - MLIR TRT

Apply Now Log in to save

As a Senior Deep Learning Software Engineer in the Solution Engineering-Automotive Machine Learning team at NVIDIA, you will develop advanced compiler technologies to optimize deep learning models for deployment on autonomous driving vehicles, leveraging NVIDIA’s unique hardware architecture. Your day-to-day responsibilities include working across various abstractions from model fine-tuning and quantization to low-level kernel development, ensuring that users can leverage frameworks like PyTorch and JAX without sacrificing performance. You will collaborate with customers to accelerate their workloads on NVIDIA platforms and stay current with the latest deep learning research to enhance Physical AI DNNs. The ideal candidate has a strong background in developing deep learning frameworks or compiler technologies, experience with GPU programming using CUDA C++, and expertise in MLIR or similar tools. This role requires solid analytical skills and a willingness to tackle complex challenges in the rapidly evolving field of autonomous vehicle technology.

Skills

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

What you'll do

  • Develop compiler technologies to optimize deep learning inference on NVIDIA hardware.
  • Work across model fine-tuning, quantization, and low-level kernel development for performance optimization.
  • Create workflows enabling users to leverage frameworks and tools without sacrificing performance.
  • Collaborate with customers to enhance their workloads on NVIDIA platforms.
  • Stay informed about deep learning research and implement new insights to improve Physical AI DNNs.

What we're looking for

  • 5+ years of software development experience
  • MS or PhD in computer science or related field with deep learning expertise
  • Experience developing deep learning frameworks or compiler technologies
  • Strong C/C++ programming skills and GPU programming knowledge
  • Familiarity with MLIR, LLVM, and other compiler technologies
  • Background in low precision inference, quantization, and DNN compression
  • Open source project ownership or significant contributions

Related searches

More Senior Software Engineer jobs Senior Software Engineer jobs in Santa Clara All jobs at Nvidia

More like this

Similar roles

Senior Deep Learning Software Engineer

Nvidia

Santa Clara, CA +1 48 days ago $224,000–$356,500
Python PyTorch JAX CUDA TensorRT NVIDIA_TensorRT_LLM GPU_optimization CUTLASS Triton Deep_learning_frameworks Performance_analysis GPU_architecture High_performance_computing Model_inference Inference_optimization
Hybrid
Save

Senior Systems Software Engineer - Deep Learning Solutions

Nvidia

Santa Clara, CA 92 days ago $224,000–$356,500
CUDA TensorRT TVM MLIR XLA Python C/C++ QNX Linux GPU Deep Learning Transformer Models Vision-Language Models Diffusion Models State Space Models Parallel Programming Memory Management Embedded Systems CI/CD MLPerf
Save

Senior Deep Learning Software Engineer, LLM Performance

Nvidia

Santa Clara, CA 56 days ago $184,000–$287,500
Python C++ CUDA TensorRT Triton PyTorch JAX TensorFlow VLLM SGLang DL compiler Performance modeling Profiling Debugging Code optimization GPU programming Deep learning framework CI/CD
Hybrid
Save

Senior Deep Learning Compiler Engineer

Nvidia

Remote (Santa Clara, CA) +1 42 days ago $152,000–$241,500
MLIR XLA TVM LLVM PyTorch CUDA C++ Python GPU CPU Embedded_Systems Cross_Compilation CI/CD
Remote
Save