Senior Software Engineer

Nvidia

Confirmed live 2 days ago High trust

Quick summary

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

Market check

Salary context

Above market

How this pay compares to similar roles

Similar $168k

This role $236k

$100k most similar roles pay here $308k

This role pays more than 98% of similar roles. Most pay $150,812–$184,900 — the shaded band above. At the midpoint, this role pays about $236k versus about $168k 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

Role Posting Log in to save

Senior Software Engineer — cuEquivariance joins the cuEquivariance team to develop a library that accelerates geometric neural networks on NVIDIA GPUs for researchers in molecular biology, materials science, and physics. The role involves building and optimizing CUDA kernels for equivariant neural network primitives like tensor products and segmented polynomials while developing Python interfaces for PyTorch and JAX. You will be responsible for the end-to-end delivery of production-quality software, managing CI/CD infrastructure for multi-GPU builds, and collaborating with research teams to translate prototypes into production kernels. Required skills include over six years of experience in CUDA and GPU programming, proficiency in C++ and Python, and a strong understanding of memory hierarchy and performance profiling. The work focuses on the technical challenge of accelerating equivariant deep learning models within scientific fields through high-performance GPU kernel engineering.

What does a Software Engineer earn in California?

Median $214000 from 775 postings across 63 companies.

See salary data

Skills

CUDA C++ Python PyTorch JAX GPU Programming Triton CI/CD Geometric Deep Learning Kernel Engineering Multi-GPU

What you'll do

Build and optimize CUDA kernels for equivariant neural network primitives to achieve peak performance across GPU generations.
Deliver end-to-end production-quality software for GPU-accelerated geometric machine learning primitives.
Develop and maintain PyTorch and JAX interfaces to expose cuEquivariance primitives to researchers and developers.
Manage CI/CD infrastructure for multi-GPU kernel builds, automated testing, and performance regression tracking.
Translate research prototypes into production kernels by collaborating with Applied Science and research teams.
Coordinate with third-party framework developers to ensure seamless integration of software into production pipelines.

What we're looking for

At least 6 years of software engineering experience with a strong background in CUDA and GPU programming.
Deep proficiency in C++ and Python for building and shipping production libraries.
Strong foundation in GPU computing including memory hierarchy, warp-level execution, occupancy, and performance profiling.
Experience developing production scientific software libraries, ML frameworks, or developer-facing GPU APIs.
Familiarity with geometric machine learning concepts such as equivariance, group representations, and tensor products.
BS/MS degree in Computer Science, Physics, Applied Mathematics, or a related field (or equivalent experience).
Preferred experience with Triton kernel development or other GPU kernel authoring tools.
Preferred advanced degree or expertise in computational chemistry, biophysics, physics, or geometric deep learning.

Save