Back to jobs

Staff Software Engineer, Machine Learning Compiler, Google Research
Posted Yesterday
Job Description
- Build, release, and support Clang/LLVM/MLIR-based compilers, debuggers, simulators, and performance analysis tools for Google TPU accelerators.
- Optimize the TPU software ecosystem to support conventional High-Performance Computing (HPC) using C, C++, and Compute Unified Device Architecture (CUDA).
- Write TPU-specific backend code generation and optimization components, ensuring support for new hardware designs.
- Create MLIR components to translate and optimize OpenHLO, CUDA, and PyTorch programs for efficient TPU execution.