TPU Architect, Silicon

1 Week ago • 3 Years + • Research & Development

Job Summary

Job Description

This role involves contributing to the innovation behind Google's hardware, focusing on Tensor Processing Units (TPUs). Responsibilities include building tools and dashboards for TPU power/performance analysis, analyzing machine learning workloads to propose architecture or compiler improvements, optimizing TPU micro-architecture, and collaborating with cross-functional teams to enhance end-to-end workload analysis flows. The ideal candidate will have experience with computer architecture, machine learning accelerators, and compiler optimization. This is a crucial role in shaping the next generation of hardware experiences at Google.
Must have:
  • Bachelor's degree in relevant field
  • 3+ years experience in computer architecture
  • Experience with microarchitecture, cache hierarchy, pipelining
  • Analyze ML workloads, propose architecture/compiler improvements
Good to have:
  • Master's/PhD in EE, CE, or CS
  • Experience with Machine Learning Accelerators
  • Experience in Machine Learning algorithms
  • Experience in architecting and optimizing compilers
  • Knowledge of compiler flows (e.g., TensorFlow)

Job Details


Minimum qualifications:

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

Preferred qualifications:

  • Master's degree or PhD in Electrical Engineering, Computer Engineering or Computer Science, with computer architecture.
  • Experience with Machine Learning Accelerators (e.g., Machine Learning Software models or accelerator architectures).
  • Experience in Machine Learning algorithms (e.g., recommendation systems, Natural Language Processing (NLP), image).
  • Experience in architecting and optimizing compilers.
  • Knowledge of compiler flows (e.g., TensorFlow).

About the job

Be part of a 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

  • Build up tools, flows, and dashboards for Tensor Processing Unit (TPU) power/performance analysis.
  • Analyze important Machine Learning workloads, evaluate power and performance and propose architecture or compiler improvements.
  • Analyze micro-architecture of the TPU, engage with the implementation team and propose power or performance optimization opportunities.
  • Collaborate with cross-functional teams to improve the end to end workload analysis flows, including debugging and tracing.

Similar Jobs

Google - Field Solutions Architect, Generative AI, Google Cloud

Google

Madrid, Community Of Madrid, Spain (On-Site)
1 Week ago
Trend Micro - NLP / Prompt Engineer (VicOne_Automotive Security)

Trend Micro

Taipei City, Taiwan (On-Site)
6 Months ago
ByteDance - Research Engineer Graduate (Machine Learning Sys-US) - 2024 Start (PhD)

ByteDance

San Jose, California, United States (On-Site)
5 Months ago
Ubisoft - Research Student - Ubisoft La Forge

Ubisoft

Shanghai, Shanghai, China (On-Site)
5 Months ago
ByteDance - Senior Software Engineer - Generative AI

ByteDance

San Jose, California, United States (On-Site)
2 Months ago
NVIDIA - Director of Mechanical Engineering

NVIDIA

Santa Clara, California, United States (On-Site)
2 Weeks ago
Google - PHY Design IP Integration Technologist

Google

Sunnyvale, California, United States (On-Site)
1 Week ago
Google - Software Engineer, Embedded Systems, Runtime, Silicon

Google

New Taipei, New Taipei City, Taiwan (On-Site)
1 Week ago
NVIDIA - Physical Design Engineer

NVIDIA

Yokne'am Illit, North District, Israel (On-Site)
3 Months ago
NVIDIA - Senior Engineering Manager, Cloud Service Provider Application

NVIDIA

Santa Clara, California, United States (On-Site)
5 Days ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

ByteDance - Tech Lead Machine Learning Engineer

ByteDance

Seattle, Washington, United States (On-Site)
1 Month ago
Google - Software Engineer, Research, Computational Imaging

Google

Mountain View, California, United States (On-Site)
4 Days ago
ByteDance - Software Engineer Large Model System Graduate (Machine Learning Sys-US) - 2024 Start (BS/MS)

ByteDance

Seattle, Washington, United States (On-Site)
5 Months ago
Google - Data and Analytics Consultant, Google Cloud

Google

Hyderabad, Telangana, India (On-Site)
6 Days ago
Meta - Software Engineer, Machine Learning

Meta

San Francisco, California, United States (On-Site)
5 Months ago
Meta - Software Engineer, Machine Learning

Meta

Pittsburgh, Pennsylvania, United States (On-Site)
5 Months ago
ION - Data Engineer, Italy

ION

Italy (Hybrid)
6 Months ago
The Walt Disney Company - Principal Machine Learning Engineer, AI Architecture - Ad Platforms

The Walt Disney Company

Santa Monica, California, United States (On-Site)
2 Days ago
Google - Senior Software Engineer, Machine Learning, Applied AI

Google

Sunnyvale, California, United States (On-Site)
1 Week ago
Twitch - Sr. Applied Scientist

Twitch

San Francisco, California, United States (On-Site)
4 Weeks ago

Get notifed when new similar jobs are uploaded

Jobs in New Taipei, New Taipei City, Taiwan

NVIDIA - Data Scientist

NVIDIA

Taipei City, Taiwan (On-Site)
2 Months ago
Trend Micro - (Sr.) Software Engineer in Linux

Trend Micro

Taipei City, Taiwan (On-Site)
6 Months ago
Corsair - Senior Hardware Development Engineer

Corsair

Taipei City, Taiwan (On-Site)
1 Month ago
NVIDIA - Senior System Software Engineer, GPU

NVIDIA

Taipei City, Taiwan (On-Site)
2 Days ago
Trend Micro - Sr. Data Scientist (AI Lab)

Trend Micro

Taipei City, Taiwan (On-Site)
6 Months ago
Google - Staff CPU RTL Design Engineer

Google

New Taipei, New Taipei City, Taiwan (On-Site)
1 Week ago
Corsair - Shipping Discrepancy Analyst

Corsair

Taoyuan City, Taiwan (On-Site)
1 Month ago
NVIDIA - System Software Engineer, GPU - New College Graduate

NVIDIA

Taipei City, Taiwan (On-Site)
1 Week ago
Google - Senior Bluetooth Firmware Engineer

Google

New Taipei, New Taipei City, Taiwan (On-Site)
6 Days ago
Appier - Software Engineer, System Integration

Appier

Taipei City, Taiwan (On-Site)
5 Months ago

Get notifed when new similar jobs are uploaded

Research & Development Jobs

The Walt Disney Company - Mechanical Designer, CAD Designer

The Walt Disney Company

Shanghai, Shanghai, China (On-Site)
2 Months ago
Tesla - Jr. SPS Programmer

Tesla

Rhineland-Palatinate, Germany (On-Site)
2 Months ago
Fluence - Controls Software Engineer-II(m/f/d)

Fluence

Erlangen, Bavaria, Germany (Hybrid)
6 Months ago
Riot Games - Manager, Learning and Development

Riot Games

United States (On-Site)
1 Month ago
Krafton  - [Publishing] Lead of Game PR (5~10년)

Krafton

Seoul, South Korea (On-Site)
3 Months ago
NVIDIA - Mixed Signal Analog Circuit Designer (RDSS Intern)

NVIDIA

Taipei City, Taiwan (On-Site)
1 Month ago
Netflix - Machine Learning Engineer (L4/5) - Studio Media Algorithms

Netflix

United States (Remote)
1 Week ago
Visa - Chief Systems Architect

Visa

Auckland, Auckland, New Zealand (Hybrid)
4 Months ago
NVIDIA - Senior Physical Design Methodology Engineer

NVIDIA

Santa Clara, California, United States (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.

Fremont, California, United States (On-Site)

Mountain View, California, United States (On-Site)

Bengaluru, Karnataka, India (On-Site)

Dublin, County Dublin, Ireland (On-Site)

Atlanta, Georgia, United States (On-Site)

Ho Chi Minh City, Ho Chi Minh City, Vietnam (On-Site)

Seattle, Washington, United States (On-Site)

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