Senior Software Engineer, JAX
Design and implement core components of the JAX framework to optimize deep learning performance on NVIDIA hardware. Collaborate with AI researchers to bridge cutting-edge numerical computing research with real-world applications. Develop tools that enhance efficiency for teams building AI systems, focusing on modularity, speed, and scalability.