Machine Learning System Tooling Tech Lead, Silicon

2 Months ago • 5 Years + • Artificial Intelligence • Research & Development

Job Summary

Job Description

This role involves designing, developing, and maintaining tools and infrastructure for analyzing Machine Learning (ML) workloads and hardware performance at Google. Responsibilities include developing power and performance models, creating visualizations and dashboards, building benchmarks for workload analysis, and collaborating with cross-functional teams. The ideal candidate possesses a Bachelor's degree (Master's or Ph.D. preferred) in a relevant field, 5+ years of experience with computer architecture concepts, and experience with ML accelerators and tooling development. The position requires expertise in compiler flows and translating high-level languages (like TensorFlow) into hardware instructions. This role is crucial for shaping the next generation of hardware experiences for Google's direct-to-consumer products, delivering unparalleled performance, efficiency, and integration.
Must have:
  • Bachelor's degree in relevant field
  • 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/Ph.D. 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 - Cloud Solutions Architect, Rapid Innovation Team

Google

(On-Site)
3 Months ago
Neostella - Machine Learning Engineer

Neostella

Medellín, Antioquia, Colombia (On-Site)
3 Months ago
Trendyol - Data Science Professionals  (LLM & Core NLP)

Trendyol

İstanbul, İstanbul, Türkiye (Hybrid)
2 Months ago
Google - Software Engineer, Pixel Camera 3A

Google

(On-Site)
3 Months ago
Madison Logic - Sr. Data Scientist

Madison Logic

Pune, Maharashtra, India (On-Site)
4 Months ago
Scale AI - Software Engineer, GenAI Model Evaluation

Scale AI

San Francisco, California, United States (Hybrid)
4 Months ago
PwC - Conversational AI Architect-Senior Associate

PwC

Bengaluru, Karnataka, India (On-Site)
4 Months ago
Wargaming - Solutions Architect (AI Technology)

Wargaming

Prague, Prague, Czechia (Hybrid)
3 Months ago
Scale AI - Army Acquisitions Writing Expert

Scale AI

St. Louis, Missouri, United States (Remote)
4 Months ago
Google - Staff Software Engineer, Platform-Aware AutoML

Google

Sunnyvale, California, United States (On-Site)
2 Months ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Google - Cloud Solutions Architect, Rapid Innovation Team

Google

Reston, Virginia, United States (On-Site)
3 Months ago
Google - Software Engineer, Auto Focus, Pixel Camera

Google

(On-Site)
3 Months ago
Autodesk - Full Stack Software Developer, Generative AI

Autodesk

Toronto, Ontario, Canada (Hybrid)
4 Months ago
Codeninja - Senior Python Developer with DevOps Expertise

Codeninja

Lahore, Punjab, Pakistan (On-Site)
4 Months ago
Activision - 2025 US Summer Internship - Data Analytics & Data Science

Activision

Irvine, California, United States (On-Site)
66 Years ago
eBay - ML Staff Software Engineer - Risk

eBay

Austin, Texas, United States (Hybrid)
4 Months ago
Frost & Sullivan - AI Engineer

Frost & Sullivan

Tamil Nadu, India (On-Site)
4 Months ago
Google - Data and Analytics Engineer

Google

(On-Site)
2 Months ago
Meta - Research Scientist Intern, Synthetic Image Generation (PhD)

Meta

Pittsburgh, Pennsylvania, United States (On-Site)
3 Months ago
ByteDance - Research Scientist in Large Model System

ByteDance

Seattle, Washington, United States (On-Site)
3 Months ago

Get notifed when new similar jobs are uploaded

Jobs in Worldwide

Looks like we're out of matches

Set up an alert and we'll send you similar jobs the moment they appear!

Artificial Intelligence Jobs

Google - Software Engineer III, AI/ML, YouTube

Google

Mountain View, California, United States (On-Site)
2 Months ago
Google - PhD Software Engineer, PhD, Early Career, Embedded Systems and Firmware, 2025 start, MSCA

Google

Sunnyvale, California, United States (On-Site)
2 Months ago
Honeywell - Lead Data Scientist-Vision AI

Honeywell

Bengaluru, Karnataka, India (Hybrid)
3 Months ago
Google - Staff Software Engineer, Google Compute Engine, Telemetry Insights

Google

Seattle, Washington, United States (On-Site)
2 Months ago
KPIT - CTO_ML/DL Data scientist

KPIT

Pune, Maharashtra, India (On-Site)
3 Months ago
Google - Software Engineer, Computational Videography and ML, Pixel Camera

Google

New Taipei, New Taipei City, Taiwan (On-Site)
2 Months ago
Hyper Verge - Machine Learning Engineer II

Hyper Verge

Bengaluru, Karnataka, India (On-Site)
4 Months ago
PwC - CCaaS (Gen AI & LLM Developer)-Senior Associate

PwC

Bengaluru, Karnataka, India (On-Site)
4 Months ago
PlayStation Global - Machine Learning Engineer II

PlayStation Global

London, England, United Kingdom (On-Site)
3 Months 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