Back to jobs
Dahl Consulting

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.

See Your Match Score

Sign up and Renata will show you how this job matches your skills and experience.

Get Started Free
Staff Software Engineer, Machine Learning Compiler, Google Research at Dahl Consulting | Renata