Senior Software Engineer, JAX
You will play a key role in NVIDIA's JAX development, designing and implementing core components to optimize performance on NVIDIA products. You'll collaborate with AI researchers and leaders to build efficient, future-proof models and tools that bridge the gap between numerical computing and real-world applications.