Ml Compiler Software Engineer, Tensor Processing Unit

Google

Sunnyvale, CA, United States
Base salary only; bonus/equity: not specified; ben...
**
Compiler parallelization features
Tpu backend optimization
Collective operation lowering
** Google is seeking a Machine Learning Compiler Software Engineer to enhance the capabilities of its Accelerated Linear Algebra (XLA) compiler, focusing on optimization and performance for large-scale ML workloads. The role involves collaboration with experts in ML hardware and compiler optimization to ensure the best performance for essential ML programs. **

Job Summary

  • This role focuses on scaling the capabilities of the XLA compiler to support increasingly large ML workloads.
  • You will deliver compiler parallelization features and optimization techniques specifically for the TPU back-end.
  • The position requires collaborating with world experts in ML hardware, compilers, and performance optimization.

Matching Summary

Match Score: 75

** Google is seeking a Machine Learning Compiler Software Engineer to enhance the capabilities of its Accelerated Linear Algebra (XLA) compiler, focusing on optimization and performance for large-scale ML workloads. The role involves collaboration with experts in ML hardware and compiler optimization to ensure the best performance for essential ML programs. **

Salary

Base salary only; Bonus/Equity: Not specified; Benefits: Not specified

Skills & Requirements

Must-have

  • compiler parallelization features
  • TPU backend optimization
  • collective operation lowering
  • horizontal scaling performance
  • ML hardware collaboration

Nice-to-have

  • debugging tool development
  • engineering experience improvement
  • roofline performance achievement

Key Requirements

  • Experience with compiler stack development
  • Knowledge of TPU architecture features
  • Background in ML performance optimization

Work Rights

Not specified

Tailored Resume

Cover Letter