Senior Software Engineer, JAX
Design and implement core components of JAX to optimize performance on NVIDIA's AI platforms, particularly around deep learning frameworks. Collaborate with researchers and engineers to bridge cutting-edge numerical computing research with real-world applications. Develop tools that enhance efficiency in AI system development, focusing on modularity, speed, and scalability.