Machine Learning System Tooling Tech Lead, Silicon

2 Months ago β€’ 5 Years + β€’ Artificial Intelligence β€’ Research & Development

Job Summary

Job Description

As a Machine Learning System Tooling Tech Lead at Google, you will design, develop, and maintain tools and infrastructure for analyzing ML workloads and hardware performance. Responsibilities include developing power and performance models, creating visualizations and dashboards, building models and benchmarks for workload analysis to inform architectural decisions. You'll collaborate with cross-functional teams to improve workload analysis flows, focusing on debuggability and tracing. This role requires expertise in computer architecture, ML accelerators, and tooling development for power, performance, and architecture analysis. A strong understanding of compiler flows and translating high-level languages (like TensorFlow) to hardware instructions is crucial. You will be part of a team developing custom silicon solutions for Google's direct-to-consumer products.
Must have:
  • 5+ years experience with computer architecture
  • Experience with ML accelerators
  • Tooling development for power/performance analysis
  • Develop and maintain performance models
  • Collaborate with cross-functional teams
Good to have:
  • Master's or PhD in performance evaluation for ML systems
  • Experience writing ML algorithms
  • Experience in architecting and optimizing compilers
  • Understanding of compiler flows

Job Details


Minimum qualifications:

  • Bachelor's degree in Electrical Engineering, Computer Engineering, Computer Science, a related field, or equivalent practical experience.
  • 5 years of experience with computer architecture concepts, including microarchitecture, cache hierarchy, pipelining, and memory subsystems.

Preferred qualifications:

  • Master's Degree or Ph.D. with an emphasis on performance evaluation for Machine Learning (ML) systems.
  • Experience with ML accelerators (e.g. having worked on ML software models or accelerator architectures).
  • Experience writing ML algorithms for e.g. recommendation systems, Natural Language Processing (NLP), image and vision.
  • Experience in tooling development for power, performance and architecture analysis.
  • Experience in architecting and optimizing compilers.
  • Understanding of compiler flows, software involved in translating a high-level language (e.g. TensorFlow) to hardware instructions.

About the job

Be part of a diverse team that pushes boundaries, developing custom silicon solutions that power the future of Google's direct-to-consumer products. You'll contribute to the innovation behind products loved by millions worldwide. Your expertise will shape the next generation of hardware experiences, delivering unparalleled performance, efficiency, and integration.

Google's mission is to organize the world's information and make it universally accessible and useful. Our team combines the best of Google AI, Software, and Hardware to create radically helpful experiences. We research, design, and develop new technologies and hardware to make computing faster, seamless, and more powerful. We aim to make people's lives better through technology.

Responsibilities

  • Design, develop, and maintain tools and infrastructure for analyzing Machine Learning (ML) workloads and hardware performance.
  • Develop and maintain power and performance models.
  • Develop visualizations and dashboards to effectively communicate performance insights to engineers.
  • Build models, benchmarks for workload analysis and help to drive architectural decisions.
  • Collaborate with cross-functional teams to improve the workload analysis flows, including debuggability and tracing.

Similar Jobs

Google - Software Engineer, Machine Learning, Google Cloud

Google

Bengaluru, Karnataka, India (On-Site)
β€’ 3 Months ago
ByteDance - Software Engineer, ML System Scheduling

ByteDance

Seattle, Washington, United States (On-Site)
β€’ 3 Months ago
ByteDance - Senior GPU System Engineer - Seattle

ByteDance

Seattle, Washington, United States (On-Site)
β€’ 3 Months ago
Blizzard Entertainment - 2025 US Summer Internship - Data Analytics & Data Science

Blizzard Entertainment

Irvine, California, United States (On-Site)
β€’ 3 Months ago
Rackspace Technology - Principal MLOPs Engineer (Canada)

Rackspace Technology

Toronto, Ontario, Canada (Remote)
β€’ 4 Months ago
Google - Software Engineer III, AI/ML, Google Research

Google

Mountain View, California, United States (On-Site)
β€’ 2 Months ago
Google - Senior Software Engineer, Machine Learning, Google Cloud Compute

Google

Sunnyvale, California, United States (On-Site)
β€’ 3 Months ago
Inworld AI - Forward Deployed Engineer - Canada

Inworld AI

Vancouver, British Columbia, Canada (Remote)
β€’ 4 Months ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

ByteDance - Machine Learning Engineer - Machine Learning Infrastructure

ByteDance

Seattle, Washington, United States (On-Site)
β€’ 3 Months ago
Intel Corporation - College Graduates

Intel Corporation

(On-Site)
β€’ 2 Months ago
Visa - Data Engineer - Sr. Consultant

Visa

Bengaluru, Karnataka, India (Hybrid)
β€’ 3 Months ago
Meta - Research Scientist, Machine Learning (PhD)

Meta

Menlo Park, California, United States (On-Site)
β€’ 3 Months ago
ByteDance - Software Development Engineer - Machine Learning System

ByteDance

Seattle, Washington, United States (On-Site)
β€’ 3 Months ago
ByteDance - Engineering Manager - Applied Machine Learning Algorithm

ByteDance

San Jose, California, United States (On-Site)
β€’ 3 Months ago
Google - Software Engineer, Auto Focus, Pixel Camera

Google

New Taipei, New Taipei City, Taiwan (On-Site)
β€’ 2 Months ago
ByteDance - Research Scientist in Foundation Model, Speech & Audio Graduates - 2024 Start (PhD)

ByteDance

Seattle, Washington, United States (On-Site)
β€’ 3 Months ago
ByteDance - Senior Algorithm Engineer - Enterprise Solution RD - San Jose

ByteDance

San Jose, California, United States (On-Site)
β€’ 2 Months ago
ByteDance - AI Security Researcher - Security - San Jose

ByteDance

San Jose, California, United States (On-Site)
β€’ 3 Months ago

Get notifed when new similar jobs are uploaded

Jobs in New Taipei, New Taipei City, Taiwan

Logitech - Software Program Manager

Logitech

Hsinchu, Hsinchu City, Taiwan (On-Site)
β€’ 4 Months ago
Rivos - Data Parallel Accelerator Performance Intern

Rivos

Hsinchu, Hsinchu City, Taiwan (Hybrid)
β€’ 4 Months ago
Logitech - Taiwan Innovation Intern

Logitech

Hsinchu, Hsinchu City, Taiwan (On-Site)
β€’ 3 Months ago
Trend Micro - Sr. Software Engineer for Networks

Trend Micro

Taipei City, Taiwan (On-Site)
β€’ 4 Months ago
Logitech - Education Team Technical Project Manager

Logitech

Hsinchu City, Taiwan (Hybrid)
β€’ 2 Months ago
Appier - Software Engineer, Machine Learning Platform

Appier

Taipei City, Taiwan (On-Site)
β€’ 3 Months ago
Google - Software Engineer, University Graduate, 2025

Google

New Taipei City, Taiwan (On-Site)
β€’ 2 Months ago
Appier - Strategic Pricing Manager

Appier

Taipei City, Taiwan (On-Site)
β€’ 3 Months ago
Google - Software Engineer, Google Pixel Camera

Google

New Taipei, New Taipei City, Taiwan (On-Site)
β€’ 2 Months ago

Get notifed when new similar jobs are uploaded

Artificial Intelligence Jobs

Google - Senior Technical Writer, Artificial Intelligence (AI)

Google

(On-Site)
β€’ 3 Months ago
Paypal - Sr. Manager, AI Tech Product Manager

Paypal

San Jose, California, United States (On-Site)
β€’ 4 Months ago
Google - Software Engineer, Auto Focus, Pixel Camera

Google

New Taipei, New Taipei City, Taiwan (On-Site)
β€’ 2 Months ago
ByteDance - Product Solution Architect, Volcano ARK (Singapore)

ByteDance

Singapore (On-Site)
β€’ 4 Months ago
Warner Bros Discovery - Senior Data Scientist

Warner Bros Discovery

Bellevue, Washington, United States (On-Site)
β€’ 2 Months ago
Zoox - Senior/Staff Software Engineer - Simulation Traffic & Behavior Modeling

Zoox

Seattle, Washington, United States (Hybrid)
β€’ 4 Months ago
DEVOTEAM - Data Driven | MLOps Engineer

DEVOTEAM

(Remote)
β€’ 4 Months ago
Zoox - Staff Software Engineer - Perception

Zoox

Foster City, California, United States (Hybrid)
β€’ 4 Months ago
Barbaricum - Senior Technical Project Manager

Barbaricum

Springfield, Virginia, United States (On-Site)
β€’ 4 Months ago
Microsoft - Senior Researcher – Generative AI – Microsoft Research AI Frontiers

Microsoft

Redmond, Washington, United States (On-Site)
β€’ 1 Month ago

Get notifed when new similar jobs are uploaded

About The Company

A problem isn't truly solved until it's solved for all. Googlers build products that help create opportunities for everyone, whether down the street or across the globe. Bring your insight, imagination and a healthy disregard for the impossible. Bring everything that makes you unique. Together, we can build for everyone.

View All Jobs

Get notified when new jobs are added by Google

Level Up Your Career in Game Development!

Transform Your Passion into Profession with Our Comprehensive Courses for Aspiring Game Developers.

Job Common Plug