Staff ML Compiler Engineer, TPU Performance Optimizations

About the role

Google Cloud's software engineers develop the next-generation technologies that change how billions of users connect, explore, and interact with information and one another.

What you'll do

• Research and develop the novel compiler optimizations targeting ML workloads, and emerging architectures
• Identify opportunities to improve the efficiency of the ML workloads through insightful performance debugging for ML workloads and custom kernels
• Provide technical leadership and mentorship as a Team Lead (TL), and explore strategic initiatives
• If you have a disability or special need that requires accommodation, please let us know by completing our Accommodations for Applicants form

What you need

• 8–14 yrs of experience
• Machine learning
• Python
• C++
• PyTorch
• NLP
• LLMs
• Any graduate

Good to have

• System design
• GCP