Senior Manager, Software Engineering - JAX at NVIDIA

Santa Clara, CA, USA · JAX is the framework of choice for the most ambitious AI research, and the datacenters it runs on are evolving fast. NVIDIA supercomputers are becoming more heterogeneous combining GPUs, CPUs, and LPUs, and JAX needs to evolve to perform well across all of them. NVIDIA seeks a Se…