Machine Learning Hardware Architect

4 Hours ago • 3-7 Years • Artificial Intelligence • $156,000 PA - $229,000 PA

Job Summary

Job Description

This role involves shaping the future of AI/ML hardware acceleration by developing cutting-edge TPU (Tensor Processing Unit) technology. You'll work with hardware and software architects to model, analyze, and define next-generation TPUs. Responsibilities include ML workload characterization and benchmarking, performance and power analysis, developing architectural models, collaborating on hardware/software co-design, and proposing optimizations for next-generation TPUs. You'll contribute to the innovation behind Google products and leverage your design and verification expertise to verify complex digital designs, focusing on TPU architecture and its integration within AI/ML systems.
Must have:
  • Bachelor's degree in relevant field or equivalent experience
  • 3+ years experience in computer architecture performance analysis and optimization
  • Experience developing software systems in C++
  • Work on ML workload characterization and benchmarking
  • Conduct performance and power analysis
Good to have:
  • 7+ years experience in computer architecture performance analysis and optimization
  • Experience in applying computer architecture principles to solve open-ended problems
  • Experience in analyzing workload performance and creating benchmarks
  • Experience in hardware and software co-design
  • Knowledge of processor or accelerator designs and mapping ML models to hardware architectures
  • Knowledge of RTL design using Verilog
Perks:
  • Bonus
  • Equity
  • Benefits

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 in computer architecture performance analysis and optimization, or a PhD degree.
  • Experience in developing software systems in C++.

Preferred qualifications:

  • 7 years of experience in computer architecture performance analysis and optimization.
  • Experience in applying computer architecture principles to solve open-ended problems.
  • Experience in analyzing workload performance and creating benchmarks.
  • Experience in hardware and software co-design.
  • Knowledge of processor design or accelerator designs and mapping ML models to hardware architectures.
  • Knowledge of design of digital reasoning at the Register Transfer Level (RTL) using Verilog.

About the job

In this role, you’ll work to shape the future of AI/ML hardware acceleration. You will have an opportunity to drive cutting-edge TPU (Tensor Processing Unit) technology that powers Google's most demanding AI/ML applications. You’ll be part of a diverse team that pushes boundaries, developing custom silicon solutions that power the future of Google's TPU. You'll contribute to the innovation behind products loved by millions worldwide, and leverage your design and verification expertise to verify complex digital designs, with a specific focus on TPU architecture and its integration within AI/ML-driven systems.

In this role, you will work with hardware and software architects to model, analyze, and define next-generation Tensor Processing Unit (TPUs).

The ML, Systems, & Cloud AI (MSCA) organization at Google designs, implements, and manages the hardware, software, machine learning, and systems infrastructure for all Google services (Search, YouTube, etc.) and Google Cloud. Our end users are Googlers, Cloud customers and the billions of people who use Google services around the world.

We prioritize security, efficiency, and reliability across everything we do - from developing our latest TPUs to running a global network, while driving towards shaping the future of hyperscale computing. Our global impact spans software and hardware, including Google Cloud’s Vertex AI, the leading AI platform for bringing Gemini models to enterprise customers.

The US base salary range for this full-time position is $156,000-$229,000 + bonus + equity + benefits. Our salary ranges are determined by role, level, and location. Within the range, individual pay is determined by work location and additional factors, including job-related skills, experience, and relevant education or training. Your recruiter can share more about the specific salary range for your preferred location during the hiring process.

Please note that the compensation details listed in US role postings reflect the base salary only, and do not include bonus, equity, or benefits. Learn more about .

Responsibilities

  • Work on Machine Learning (ML) workload characterization and benchmarking.
  • Conduct performance and power analysis and evaluate proposals.
  • Develop architectural and microarchitectural models to enable quantitative analysis.
  • Collaborate with partners in hardware design, software, compiler, Machine Learning (ML) model and research teams for hardware/software co-design.
  • Propose capabilities and optimizations for next-generation TPUs and chip roadmap.

Similar Jobs

Bungie - Marathon Global Community Lead

Bungie

United States (Hybrid)
4 Weeks ago
Ziff Davis - Senior Software Engineer, Backend - Lose It!

Ziff Davis

United States (On-Site)
5 Months ago
Google - ASIC Design Engineer, Platform IP, Silicon

Google

Mountain View, California, United States (On-Site)
1 Day ago
Volley - Director of Engineering, Games

Volley

San Francisco, California, United States (Hybrid)
2 Months ago
Google - Senior Software Engineer, Full Stack, Core

Google

Hyderabad, Telangana, India (On-Site)
1 Day ago
Meta - Software Engineer, Machine Learning

Meta

United States (Remote)
1 Day ago
ByteDance - Research Scientist, Vision Foundation Model

ByteDance

San Jose, California, United States (On-Site)
5 Months ago
Google - Software Engineer III, AI/ML Recommendations, Rankings, Predictions, Google Cloud

Google

Hyderabad, Telangana, India (On-Site)
1 Day ago
ByteDance - Research Engineer (Foundation Model) - Machine Learning Systems

ByteDance

Singapore (On-Site)
5 Months ago
Meta - Software Engineer, Machine Learning

Meta

Burlingame, California, United States (On-Site)
5 Months ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

PwC - Senior Consultant - Technology Consulting - Financial Sector

PwC

Prague, Prague, Czechia (On-Site)
6 Months ago
Scanline VFX - Art Director

Scanline VFX

Seoul, South Korea (On-Site)
10 Months ago
Crunchyroll - People Services Coordinator

Crunchyroll

Dallas, Texas, United States (On-Site)
2 Months ago
ComeOn Group - QA Engineer

ComeOn Group

Styria, Austria (Hybrid)
4 Weeks ago
Tencent - Investment Analyst Intern

Tencent

California, United States (On-Site)
1 Month ago
Blinkhealth - Manager, Marketing Operations

Blinkhealth

(Remote)
2 Months ago
PlayStation Global - Senior Program Manager, Account & Identity

PlayStation Global

California, United States (On-Site)
3 Weeks ago
Light Speed Studios - Global Publishing PM (Japan)

Light Speed Studios

Tokyo, Japan (On-Site)
1 Month ago
Evolution - In Studio Game Presenter

Evolution

Atlantic City, New Jersey, United States (On-Site)
2 Months ago
Anthology  Inc  - Marketing Research and Data Analyst

Anthology Inc

Bogotá, Bogota, Colombia (Remote)
3 Weeks ago

Get notifed when new similar jobs are uploaded

Jobs in Sunnyvale, California, United States

IGT - SAP ABAP Developer

IGT

Reno, Nevada, United States (On-Site)
4 Months ago
Trackman - Trackman Baseball System Operator

Trackman

Arizona, United States (On-Site)
3 Weeks ago
Meta - Global Sales Analytics Lead

Meta

New York, New York, United States (Remote)
5 Months ago
Next Level Business Services - SAP SM and MM (Full Time)

Next Level Business Services

Malvern, Pennsylvania, United States (On-Site)
6 Months ago
Google - PHY Design IP Integration Technologist, Machine Learning

Google

Sunnyvale, California, United States (On-Site)
3 Hours ago
Next Level Business Services - .NET Developer

Next Level Business Services

Minneapolis, Minnesota, United States (On-Site)
6 Months ago
DraftKings - Senior Social Marketing Associate - Creator

DraftKings

United States (Remote)
1 Week ago
Rackspace Technology - Government Services Sales Executive VI

Rackspace Technology

Sacramento, California, United States (Remote)
2 Months ago
Netflix - Software Engineer (L5), Ads Measurement

Netflix

New York, New York, United States (On-Site)
1 Day ago

Get notifed when new similar jobs are uploaded

Artificial Intelligence Jobs

Inworld AI - AI Trainer (Contractor) - Writing & Gaming

Inworld AI

Mountain View, California, United States (Remote)
1 Month ago
Google - Software Developer III, AI/ML, Google Cloud AI

Google

Waterloo, Ontario, Canada (On-Site)
1 Day ago
Glean - Software Engineer, Machine Learning (India)

Glean

Bengaluru, Karnataka, India (On-Site)
5 Months ago
Meta - Software Engineer, Systems ML - SW/HW Co-design

Meta

Redmond, Washington, United States (On-Site)
5 Months ago
Ubisoft - Senior ML Data Scientist

Ubisoft

Montreal, Quebec, Canada (On-Site)
3 Weeks ago
Trend Micro - Sr. AI Engineer

Trend Micro

Taipei City, Taiwan (On-Site)
6 Months ago
GoTo Group - Senior Data Scientist - Computer Vision - KYC

GoTo Group

Singapore (On-Site)
6 Months ago
Google - Senior Staff Software Engineer, Accelerators Systems Software Architect

Google

Sunnyvale, California, United States (On-Site)
1 Day ago
SmileGate - AI Voice Model Developer

SmileGate

Seongnam-si, Gyeonggi-do, South Korea (On-Site)
3 Weeks ago
Passive Logic - AI Control Theory & Optimization Scientist

Passive Logic

Salt Lake City, Utah, United States (On-Site)
4 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.

Seoul, South Korea (On-Site)

Tel Aviv-Yafo, Tel Aviv District, Israel (On-Site)

Hyderabad, Telangana, India (On-Site)

Atlanta, Georgia, United States (On-Site)

Fremont, California, United States (On-Site)

Milan, Lombardy, Italy (On-Site)

Eemshaven, Groningen, Netherlands (On-Site)

Bengaluru, Karnataka, India (On-Site)

Sunnyvale, California, 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