Senior Software Engineer, JAX ML Systems
Listed on 2026-10-17
-
Software Development
Senior Developer, Software Engineer, AI Engineer (Applied/Software), Machine Learning/ ML Engineer
NVIDIA Corporation in Santa Clara, CA is seeking a senior software engineer to advance its AI platform, focusing on performance optimizations for JAX and related deep learning tooling.
You will design core JAX components, optimize CPU/GPU paths, collaborate with AI researchers, and build scalable tools to accelerate teams shipping AI applications.
The role requires a BS in CS/CE, 6+ years of experience, strong C/C++ and Python, and exposure to major ML frameworks;
Level 4/5 compensation applies.
This opening is for the Senior Software Engineer, JAX:
Performance & ML Systems role at NVIDIA Corporation.
This posting is for the Senior Software Engineer, JAX:
Performance & ML Systems role at NVIDIA Corporation, based in Santa Clara, CA, United States.
We are looking to fill the Senior Software Engineer, JAX:
Performance & ML Systems position at NVIDIA Corporation in Santa Clara, CA, United States.
(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).