Senior Software Engineer, CUTLASS Platform

Nvidia

Confirmed live 2 days ago High trust

Quick summary

Work type
On-site
Location
Santa Clara, CAAustin, TXHillsboro, ORDurham, NCRedmond, WA
Salary
$152,000–$241,500 / yr
Posted
102 days ago
Freshness
Confirmed live 2 days ago

Market check

Salary context

Above market

How this pay compares to similar roles

Similar $182k
This role $197k
$105k most similar roles pay here $256k

This role pays more than 69% of similar roles. Most pay $151,000–$213,000 — the shaded band above. At the midpoint, this role pays about $197k versus about $182k 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 896 open roles on FindRole.

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

Most-posted roles

View all roles at Nvidia

At a glance

TL;DR · Senior Software Engineer, CUTLASS Platform

As a Senior Software Engineer, CUTLASS Platform, you will join the Deep Learning Library team to develop core components of the CUTLASS platform, an open-source ecosystem for high-performance linear algebra and Tensor Core primitives. You will build Tensor Core MMAs, copies, synchronization barriers, and schedulers using CUDA C++ and a CUTLASS Python DSL. Your work involves advancing the MLIR-based backend compiler stack by designing dialects and compiler passes while authoring example kernels to showcase novel GPU hardware features. The role requires expertise in C++, Python, and high-performance code generation, alongside knowledge of compiler transformations and optimizations. You will collaborate with architecture and compiler teams to evaluate programming models for deep learning computations. This position focuses on solving complex problems in matrix multiply operations and providing abstractions for advanced GPU hardware features.

What does a Software Engineer earn in California?

Median $214000 from 775 postings across 63 companies.

See salary data

What you'll do

  • Develop core components of the CUTLASS platform including Tensor Core MMAs and synchronization barriers in CUDA C++.
  • Build features for the CUTLASS Python DSL to enable high-performance kernel development.
  • Advance the MLIR-based backend compiler stack by designing dialects and associated compiler passes.
  • Author example kernels using CUTLASS abstractions to showcase novel GPU hardware features.
  • Provide feedback on programming models to the GPU architecture, CUDA, and NVVM/PTX compiler teams.
  • Assess the performance of future GPU hardware features for integration into the software stack.

What we're looking for

  • Master's or PhD degree in Computer Science, Computer Engineering, or a related field (or equivalent experience).
  • 3+ years of relevant industry experience.
  • Strong proficiency in C++ programming and software design, including debugging, performance evaluation, and testing.
  • Experience with high-performance code generation and knowledge of compiler transformations and optimizations.
  • Deep understanding of computer architecture and parallel computing programming models.
  • Experience writing high-performance kernels at low levels of abstraction like NVVM/PTX for GPUs or similar architectures.
  • Hands-on compiler design experience, particularly in MLIR.
  • Understanding of deep learning models, algorithms, and frameworks.

More like this

Similar roles

Senior AI Frameworks Engineer

Nvidia

Santa Clara, CA +1 106 days ago $152,000$241,500
C++ Python CUDA LLVM MLIR JIT AST FFI GPU Programming Conda High-Performance Computing DSL NVIDIA GPU
3+ yrs exp

Deep Learning Compiler Engineer

Nvidia

Remote (Santa Clara, CA) +1 19 days ago $152,000$241,500
CUDA C++ C MLIR LLVM XLA TVM Compiler Optimization GPU Architecture Performance Analysis Deep Learning IR Design OpenCL
3+ yrs exp Remote

Senior Deep Learning Compiler Engineer

Nvidia

Remote (Santa Clara, CA) +1 134 days ago $152,000$241,500
C++ Python CUDA OpenCL MLIR XLA TVM LLVM PyTorch GPU Architecture Compiler Optimization Deep Learning Kernel Generation cross compilation Performance Analysis
3+ yrs exp Remote

Senior AI Compiler Engineer, MLIR

Nvidia

Remote (Santa Clara, CA) +1 28 days ago $152,000$241,500
MLIR LLVM XLA C++ Python CUDA OpenCL PyTorch JAX GPU Architecture Kernel Generation Compiler Optimization Performance Analysis Deep Learning
3+ yrs exp Remote

Senior Software Engineer, AI Platform

Anduril Industries

Seattle, WA 60 days ago $191,000$253,000
CI/CD Python C++ Rust Terraform Ansible Docker Kubernetes GitLab CI Jenkins GitHub Actions Crossplane Puppet Chef Bash Prometheus Grafana ELK Stack Splunk GitOps