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 and engineers to bridge cutting-edge numerical computing research with real-world applications. Develop tools that enhance productivity across AI development teams, focusing on modularity, speed, and scalability.