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 to build advanced models and tools that enhance the efficiency of AI systems, bridging the gap between cutting-edge research and real-world applications.