Senior GPU Kernel Engineer — Geometric ML
Listed on 2026-10-21
-
Software Development
AI Engineer (Applied/Software), Machine Learning/ ML Engineer, Software Engineer
NVIDIA Corporation in Santa Clara, CA seeks a Senior Software Engineer for the cu Equivariance team to advance GPU-accelerated ML primitives and production libraries. You will implement CUDA kernels and maintain interfaces for PyTorch and JAX, ensuring scalable, production-ready software across NVIDIA GPU generations.
The role requires strong C++/Python, deep GPU knowledge, and collaboration with research teams and external partners.
We would love to welcome a new Senior GPU Kernel Engineer — Geometric ML to our group in Santa Clara, CA, United States.
We aim to respond to suitable candidates as soon as possible.
Full responsibilities and requirements are described in the listing above.
Learn more about the Senior GPU Kernel Engineer — Geometric ML role in the description above.
We appreciate your interest in this position.
Join NVIDIA Corporation and contribute to our ongoing work.
(If this job is in fact in your jurisdiction, then you may be using a Proxy or VPN to access this site, and to progress further, you should change your connectivity to another mobile device or PC).