- 6+ years of software engineering experience with a strong background in CUDA and GPU programming.
- Deep proficiency in C++ and Python; experience building and shipping production libraries used by external developers.
- Good foundation in GPU computing: memory hierarchy, warp-level execution, occupancy, and performance profiling methodology.
- Experience building or chipping in to production scientific software libraries, ML frameworks, or developer-facing GPU APIs.
- Familiarity with concepts in geometric machine learning — equivariance, group representations, irreducible representations, or tensor products — sufficient to work efficiently in the domain.
- BS/MS in Computer Science, Physics, Applied Mathematics, or a related field, or equivalent experience.
Ways to Stand Out from the Crowd
- You have chipped in to or deeply used a major neural network framework that respects equivariance: e3nn, MACE, NequIP, SE(3)-Transformers, or similar.
- Hands-on experience with Triton kernel development or other GPU kernel authoring tools alongside CUDA.
- Experience with mixed-precision or tensor-core-aware algorithm design for scientific or ML workloads.
- PhD or equivalent experience in computational chemistry, biophysics, physics, or computer science with a focus on geometric deep learning or HPC.
- Contributions to open-source geometric ML or GPU computing projects.
Widely considered to be one of the technology world’s most desirable employers, NVIDIA offers highly competitive salaries and a comprehensive benefits package. As you plan your future, see what we can offer to you and your family www.nvidiabenefits.com/
Your base salary will be determined based on your location, experience, and other objective job-related factors. The expected base salary ranges for this position are: 184,000 USD – 287,500 USD for Level 4; and 224,000 USD – 356,500 USD for Level 5.
You will also be eligible for equity and benefits.
Applications for this job will be accepted at least until November 3, 2026.
This posting is for an existing vacancy.
NVIDIA uses AI tools in its recruiting processes.