Senior Software Engineer, JAX
As a Senior Software Engineer, you will play a key role in NVIDIA's efforts to enhance JAX, a deep learning framework. You will design and implement core components, optimize performance on NVIDIA products, and collaborate with AI researchers to build advanced models and tools that bridge the gap between numerical computing and real-world applications.