Senior Software Engineer, ML Compilers, TPU, YouTube

Senior Software Engineer, ML Compilers, TPU, You Tube

You Tube place San Bruno, CA, USA ;
Mountain View, CA, USA

Apply

Note:

By applying to this position you will have an opportunity to share your preferred working location from the following:
San Bruno, CA, USA;
Mountain View, CA, USA
.

• 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.
• 3 years of experience with software design and architecture.
• Experience with state-of-the-art ML compilers and their internals, experience writing compiler optimization passes.
• Experience with ML frameworks such as Tensor Flow, JAX, and PyTorch, or ML compilers (e.g., accelerated linear algebra (XLA)).

• Master's degree or PhD in computer science or related technical fields.
• Experience with debugging correctness and performance issues at all levels of the ML software stack.
• Familiarity with accelerator HW architectures (TPUs/GPUs).

On this team, you will own the optimization of the models powering the You Tube algorithm. Your work will range from contributing to the Accelerated Linear Algebra (XLA) compiler for Google's custom Tensor Processing Units, to authoring custom Pallas kernels in JAX, maximizing fleet utilization and the value delivered to users.

You will build support and optimize new and existing models in our Rec Sys stack, including new model architectures while adapting to next-generation TPU hardware.

You will engage in state-of-the-art model and TPU compiler co-design, with opportunities to work up and down the stack ranging from end-user ML models down to Hardware/Software architecture.

US: $174000 - $253000 (USD) + 15% bonus target + equity + benefits

• 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 out of current and next-generation TPUs for our unique LEM (Large Embedding Models) requirements, directly impacting fleet efficiency.
• Collaborate closely with You Tube’s Next Platform Evaluation team and Google’s hardware designers to co-design future processors.

Google is proud to be an equal opportunity and affirmative action employer. We are committed to building a workforce that is representative of the users we serve, creating a culture of belonging, and providing an equal employment opportunity regardless of race, creed, color, religion, gender, sexual orientation, gender identity/expression, national origin, disability, age, genetic information, veteran status, marital status, pregnancy or related condition (including breastfeeding), expecting or parents-to-be, criminal histories consistent with legal requirements, or any other basis protected by law.

See also Google's EEO Policy, Know your rights: workplace discrimination is illegal, Belonging at Google, and How we hire.

Google is a global company and, in order to facilitate efficient collaboration and communication globally, English proficiency is a requirement for all roles unless stated otherwise in the job posting.

Equity is granted exclusively and discretionarily by Alphabet Inc. on the basis of an agreement concluded between you and Alphabet Inc. Alphabet Inc. is your sole contractual partner with respect to equity grants. GSU grants are not guaranteed, are discretionary, are subject to approval by the Alphabet Inc. board of directors or its delegate, the terms of the relevant Alphabet Inc.

stock plan, and your grant agreement. They have no impact on statutory payments. Current or past grants do not confer an acquired right.