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

Staff ML Compiler Engineer, TPU Performance Optimizations

Google · Bengaluru

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

ML Compiler Engineer, TPU Performance Optimizations

Google · Bengaluru

Minimum qualifications: Bachelor's degree or equivalent practical experience. 8 years of experience in software development. 3 years of experience with software design and architecture. Experience with machine learning,