All vacancies
Senior Software Engineer - cuEquivariance
NVIDIA
US, CA, Santa Clara · On-site · relocation$15,333–29,708 gross/mofull-timeNVIDIA Careers (offices abroad)
NVIDIA has been transforming computer graphics, PC gaming, and accelerated computing for more than 25 years. It’s a unique legacy of innovation that’s fueled by great technology—and amazing people.
Responsibilities
- Build, implement, and optimize CUDA kernels for equivariant neural network primitives — tensor products, segmented polynomials, and triangle-based operations — targeting peak performance across NVIDIA GPU generations.
- Build and maintain the interfaces for PyTorch and JAX that expose cuEquivariance primitives to application developers and researchers.
- Drive CI/CD infrastructure for multi-GPU kernel builds, automated correctness testing, and performance regression tracking.
- Collaborate with Applied Science and research teams to evaluate new equivariant architectures and translate prototypes into production kernels.
Languages
- Work format
- On-site
- Seniority
- Senior
- Posted
- 17 Aug 2026
- Last verified
- In the last 5 days
Similar roles
- Software Engineer (GitHub Contributor), Remote- ContractXperteez Technology · Remote · worldwide, UTC+4 accepted
- Senior Backend DeveloperCommit · Remote · Armenia
- Senior Dynamics CRM Developer (VoC)EPAM Systems · Remote · Armenia
- Staff Software EngineerDwelly · Remote · worldwide, UTC+4 accepted
- Mid-Senior Full Stack Engineer - RemotePixibo · Remote · worldwide, UTC+4 accepted
- Information Security Engineer/ Cybersecurity SpecialistGrovf · Remote · Armenia