Staff ML Compiler Engineer for TPU Acceleration

Google; New York (NY); On-site; USD 207,000 - 301,000; Full time

Benefits offered by this job

20% bonus target

Equity options

Comprehensive benefits

Job summary

Google is looking for a Staff Software Engineer, ML Compilers, TPU in New York, New York. In this role, you will design, develop, test, deploy, maintain, and enhance the XLA compiler, accelerating machine learning workloads for internal and external customers.

Applicants are required to have a Bachelor's degree and significant experience in programming, testing, and software design. This position offers a competitive salary of $207,000 - $301,000 + bonus and equity options.

Qualifications

• Bachelor's degree or equivalent practical experience.
• 8 years of experience programming in C++ or Python.
• 5 years of experience testing and launching software products.
• 5 years of experience with performance, large-scale systems data analysis, visualization tools, or debugging.
• 3 years of experience with software design and architecture.

Responsibilities

• Contribute to the compiler for a novel processor designed to accelerate machine learning workloads.
• Target and compile high-performance implementations of operations at distributed scale.
• Design and implement new compiler passes that extract more performance from current and next-generation TPUs, directly impacting fleet efficiency.
• Collaborate closely with hardware designers to co-design future processors.
• Research high-level representations to effectively program large-scale, distributed, and heterogeneous systems.

Skills

C++

Python

Large-scale systems data analysis

Testing and launching software products

Software design and architecture

Education

Bachelor's degree or equivalent practical experience

Job description

Google is looking for a Staff Software Engineer, ML Compilers, TPU in New York, New York. In this role, you will design, develop, test, deploy, maintain, and enhance the XLA compiler, accelerating machine learning workloads for internal and external customers.

Applicants are required to have a Bachelor's degree and significant experience in programming, testing, and software design. This position offers a competitive salary of $207,000 - $301,000 + bonus and equity options.