Google

Staff Software Engineer, Machine Learning Compiler, Google Research

Google • $207K — $301K *
Austin, TX 78745In-Person
Information Technology
8 - 10 years of experience
Today
Be an Early Applicant
Job Overview by Ladders

Qualifications

  • Bachelor's degree in Computer Science or equivalent experience.
  • 8 years of software development experience in programming languages like Python, C, C++, Java, or JavaScript.
  • 5 years of experience in compiler development or construction.
  • Experience with modifying LLVM as a contributor.
  • Master's degree or PhD in Computer Science or related field (preferred).
  • Experience with Clang/LLVM/MLIR for VLIW processors or ML accelerators (preferred).
  • Experience in designing or managing large-scale systems and HPC environments (preferred).

Responsibilities

  • Build, release, and support compilers, debuggers, simulators, and performance analysis tools for Google TPU accelerators.
  • Optimize the TPU software ecosystem for HPC using C, C++, and CUDA.
  • Develop TPU-specific backend code generation and optimizations to accommodate new hardware designs.
  • Create MLIR components to translate and optimize OpenHLO, CUDA, and PyTorch programs for TPUs.

Benefits

  • Health, dental, and vision insurance.
  • 401(k) plan with company matching.
  • Generous paid time off and sick leave.
  • Employee Assistance Program (EAP).
  • Opportunities for professional development and growth.
Full Job Description
info_outline
X Note: By applying to this position you will have an opportunity to share your preferred working location from the following: Austin, TX, USA; Mountain View, CA, USA.

Minimum qualifications:
  • Bachelor's degree in Computer Science or equivalent practical experience.
  • 8 years of experience with software development in one or more programming languages (e.g., Python, C, C , Java, JavaScript).
  • 5 years of experience with compiler development or construction.
  • Experience with low level virtual machines (LLVM) development, modifying LLVM as a contributor.

Preferred qualifications:
  • Master's degree or PhD in Computer Science, or a related technical field.
  • Experience developing compilers using the Clang/LLVM/Multi-Level Intermediate Representation (MLIR) infrastructure, specifically for Very Long Instruction Word (VLIW) processors or Machine Learning (ML) accelerators.
  • Experience in the design or management of large-scale systems and High-Performance Computing (HPC) environments.


About the job

The Machine Learning Compiler (MPACT) team in Google Research is developing open-source, retargetable compiler infrastructure in the Low Level Virtual Machines (LLVM) and MLIR frameworks to streamline iterative processor and system co-design for the Google Tensor Processing Unit (TPU) family of processors, and to enable C/C /Cuda High Performance Computing (HPC) codes to run efficiently on TPUs.

US: $207000 - $301000 (USD) 20% bonus target equity benefits

Learn more about benefits at Google .

Responsibilities
  • 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.

About Google

Google is a multinational technology company that specializes in Internet-related services and products. These include online advertising technologies, search engine, cloud computing, software, and hardware. Google was founded in 1998 by Larry Page and Sergey Brin while they were Ph.D. students at Stanford University. The company has grown tremendously since then and has become one of the most valuable companies in the world. Google's mission is to organize the world's information and make it universally accessible and useful.
Learn more about Google
Size
156,500 employees
Market Cap
$1,115.4 billion
Industry
Enterprise Technology
Net Income
$40.2 billion
Founded
1998
5 Year Trend
+23.3%
Revenue
$182.5 billion
NASDAQ
GOOGL
* Ladders Estimates

Similar Jobs

Ad banner background

Get Ready For Your
Next Interview

More Jobs at Google

More Information Technology Jobs

Find similar Staff Software Engineer, Machine Learning Compiler, Google Research jobs:

NationwideAustin, TX