Senior Software Engineer, JAX
Design and implement core components of JAX 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 improve efficiency across AI development teams, focusing on modularity, speed, and scalability.