Senior Software Engineer, JAX
Develop and optimize core components of NVIDIA's AI platform using JAX, focusing on performance, modularity, and integration with deep learning frameworks. Collaborate with researchers and engineers to bridge cutting-edge numerical computing research with real-world applications. Work across distributed systems and GPU architectures to enhance AI tooling for global research impact.