"Staff ML Compiler Engineer, TPU Performance Optimizations" Jobs in India

Responsibilities Learn and build an intuitive understanding of various parts of Google’s ML stack: Frameworks (e.g., JAX, PyTorch), XLA and runtime stack. Research and develop the novel compiler optimizations targeting M