GPU Kernel Engineer for Equivariant ML (CUDA/PyTorch
Job in
Santa Clara, Santa Clara County, California, 95053, USA
Listed on 2026-08-20
Listing for:
NVIDIA Corporation
Full Time
position Listed on 2026-08-20
Job specializations:
-
Software Development
Python, Software Engineer, AI Engineer (Applied/Software)
Job Description & How to Apply Below
NVIDIA Bio Ne Mo seeks a Senior Software Engineer to advance cu Equivariance, a GPU-accelerated library enabling equivariant models ’ll work on CUDA kernels, Python interfaces, and production pipelines for PyTorch and JAX, collaborating with researchers and external developers.
We expect 6+ years in software engineering with strong C++/Python, GPU programming, and a solid track record delivering production libraries.
#J-18808-LjbffrTo View & Apply for jobs on this site that accept applications from your location or country, tap the button below to make a Search.
(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).
(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).
Search for further Jobs Here:
×