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