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. Collaborate with AI researchers and leaders to build future-proof models and tools that enhance the efficiency of AI-based systems, bridging the gap between numerical computing and real-world applications.