company_site
Staff ML Compiler Engineer, TPU Performance Optimizations
Google
Bengaluru, IND
Senior · 8+ years experience
Discovered 2 weeks ago
cplusplusgcpjaxpythonpytorch
Free
Job Fit Check
Base Career helps you apply smarter for this job.
?%
Ready to ScanKey skills for this role
cplusplusgcpjax
Key Skills for This Role
cplusplusgcpjaxpythonpytorch
Full Job Posting
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 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, and build compiler solutions to deliver those improvements.
- Provide technical leadership and mentorship as a Team Lead (TL), and explore strategic initiatives.
- - 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 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, and build compiler solutions to deliver those improvements. - Provide technical leadership and mentorship as a Team Lead (TL), and explore strategic initiatives.
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, compilers, computer architecture, GPU programming, C++, Python.
Preferred qualifications:
Experience in open-source software development, including experience in releasing and supporting open-source projects.
Experience with ML compilers and their internals, experience writing compiler optimization passes.
Experience with debugging correctness and performance issues at all levels of the ML Software (SW) stack.
Apply for this job in 1 click
Skip the repetitive application forms
Install the Base Career Chrome Extension and autofill job applications across major job boards with your profile.
Trusted by over 500,000 job seekers on Base Career
More from this employer
More jobs at Google
Director, Trust and Safety, Trusted Experiences, Google DeepMind
Sunnyvale, USA
Director
Discovered TodayView →
Director, 1P Direct to Consumer Commerce and Performance Marketing
Mountain View, USA
Director
Discovered TodayView →
Senior Software Engineer, AI/ML GenAI, Google Cloud Data Management
Sunnyvale, USA
Senior
Discovered TodayView →
Director, Regulatory Affairs, Content Advisory
London, GBR
Discovered Today
Mixed-Methods UX Researcher
San Bruno, USA
Discovered Today
Director, Trust and Safety, Trusted Experiences, Google DeepMind
Sunnyvale, USA
Discovered Today
Director, 1P Direct to Consumer Commerce and Performance Marketing
Mountain View, USA
Discovered Today
Senior Software Engineer, AI/ML GenAI, Google Cloud Data Management
Sunnyvale, USA
Discovered Today
Research Scientist, Google Research
Mountain View, USA
Discovered Today
Analyst, Technical Infrastructure Assets
Chicago, USA
Discovered Today
Senior Analytical Lead, Large Customer Sales
San Francisco, USA
Discovered Today
