4 hours ago
Base Salary
$174k - $252k/yr
Responsibilities
- Deliver compiler parallelization features and optimization techniques for TPU backends supporting large-scale workloads.
- Contribute to collective operation lowering and implementation on the TPU platform.
- Develop low-level compiler optimization techniques throughout the compiler stack.
- Analyze TPU architecture features and use them to optimize horizontal scaling performance.
- Build compiler debugging tools to prevent scaling issues and improve engineering experience.
Requirements
- Bachelor's degree in Computer Science or a related technical field, or equivalent practical experience.
- At least 5 years of software development experience in one or more programming languages, including Python and C++.
- At least 3 years of experience with machine learning infrastructure, ML execution frameworks such as TensorFlow, JAX, or PyTorch, or hardware accelerators such as TPUs or GPUs.
- At least 2 years of experience with a low-level systems programming language such as C++.
- Experience with performance and compilers.
- Preferred: 3 years of experience in low-level ML accelerator programming, compiler programming, or close-to-hardware performance programming.
- Preferred: experience profiling workloads and implementing performance optimizations.
- Preferred: experience writing high-performance, readable C++.
- Preferred: experience with hardware design and hardware architecture.
- Preferred: working knowledge of ML compilers such as XLA or MLIR and experience co-designing hardware-aware optimizations for model execution.
Benefits
- 15% bonus target, equity, and benefits are offered in addition to base pay.
Categories
About Google
A problem isn't truly solved until it's solved for all. Googlers build products that help create opportunities for everyone, whether down the street or across the globe. Bring your insight, imagination and a healthy disregard for the impossible. Bring everything that makes you unique. Together, we can build for everyone. Check out our career opportunities at goo.gle/3DLEokh