Silicon AI/ML Architect, TPU

7 Hours ago • 10 Years + • Research & Development • Artificial Intelligence

Job Summary

Job Description

This role involves architecting AI/ML SoCs and supporting implementation and validation phases (pre-silicon and post-silicon). You will collaborate with internal and external teams to understand SoC function requirements, considering hardware, software, and economic factors. Responsibilities include developing SoC architecture and specifications, evaluating silicon solutions (off-the-shelf, co-developments, custom designs, chiplets), creating high-performance hardware/software interfaces, and collaborating with design, verification, and validation stakeholders. You will also evaluate power, performance, and area tradeoffs, driving improvements across generations. The work contributes to Google's direct-to-consumer products and the next generation of hardware experiences.
Must have:
  • 10+ years in ML/graphics SoC architecture
  • 8+ years in multi-chiplet SoC tradeoff analysis
  • ASIC design flow experience (spec to production)
  • SoC architecture and specification development
  • Collaboration with system and software teams
Good to have:
  • SoC design, RTL coding, integration flows
  • Hardware/software interface optimization
  • Performance estimation, test plan definition
  • Procedural programming (C++, Python, Go)
  • High-performance/low-power design techniques

Job Details

Minimum qualifications:

  • Bachelor's degree in Electrical Engineering, Computer Engineering, Computer Science, a related field, or equivalent practical experience.
  • 10 years of experience in architecture of machine learning or graphics SoCs.
  • 8 years of experience with performance, power, area, cost tradeoff analysis for multi chiplet SoC architectures.
  • Experience with ASIC design flows, from specification to production.

Preferred qualifications:

  • Experience in SoC designs, RTL coding and integration flows.
  • Experience working with software teams optimizing the hardware/software interface.
  • Experience in estimating performance by analysis and modeling, and experience in defining and driving performance test plans.
  • Experience in a procedural programming language (e.g., C++, Python, Go).
  • Knowledge of high performance and low power design techniques.
  • Knowledge of arithmetic units, bus architectures, accelerators, or memory hierarchies.

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.

As a Silicon AI/ML Architect, you will drive the Architecture of an AI/ML SoC and will support the engineering teams during the implementation and validation phases, pre-silicon, and post-silicon. You will work with internal system teams and external partners to develop an understanding of the required SoC functions given its surrounding hardware, software, and the economic environments. You will also collaborate with the Engineering teams to drive the optimal balance between performance, features, power, schedule, and cost.

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.

Responsibilities

  • Develop SoC architecture and Architecture specifications that meet current and future computing requirements for AI/ML roadmap.
  • Evaluate different silicon solutions for executing Google’s data center AI accelerator roadmap: off-the-shelf components, vendor co-developments, custom designs, and chiplets.
  • Collaborate with Systems and Software teams to create high performance hardware/software interfaces.
  • Collaborate with design, verification, emulation, physical design, packaging, and silicon validation stakeholders to ensure that designs are complete, correct, and performant.
  • Evaluate power, performance, area tradeoffs and drive improvements across generations.

Similar Jobs

Playrix - Senior C++ Software Engineer (Gameplay)

Playrix

Ireland (Remote)
5 Months ago
Microsoft - Senior Build Engineer

Microsoft

Ostergotland, Östergötland County, Sweden (Hybrid)
9 Hours ago
NVIDIA - Senior Firmware Design Engineer

NVIDIA

Yokne'am Illit, North District, Israel (Hybrid)
2 Months ago
Nagarro - Staff Engineer - DevOps Site Reliability

Nagarro

Colombia (Remote)
2 Months ago
Applike Group - Senior Data Scientist (Recommendation Systems Expert) (f/m/d)

Applike Group

Hamburg, Hamburg, Germany (Hybrid)
6 Months ago
NVIDIA - Senior System Software Architect, HPC Networking

NVIDIA

Tel Aviv-Yafo, Tel Aviv District, Israel (On-Site)
1 Month ago
Backbone - Electrical Engineer

Backbone

Atherton, California, United States (Hybrid)
9 Months ago
NVIDIA - Principal Software Architect, GPU Networking Research

NVIDIA

Tel Aviv-Yafo, Tel Aviv District, Israel (On-Site)
1 Month ago
NVIDIA - Senior System Software Engineer, GPU Firmware

NVIDIA

Bengaluru, Karnataka, India (On-Site)
2 Months ago
Riot Games - Senior Researcher, Wild Rift

Riot Games

Shanghai, Shanghai, China (On-Site)
3 Weeks ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Tencent - NLP Research Intern 104493

Tencent

London, England, United Kingdom (On-Site)
3 Months ago
Tencent - Tencent Cloud - Senior Cloud Network Engineer

Tencent

(On-Site)
1 Month ago
Amazon Games - Senior ML Scientist, Amazon Games AI Research

Amazon Games

San Diego, California, United States (On-Site)
3 Months ago
Genies - Machine Learning Engineer: 3D Generative AI

Genies

San Mateo, California, United States (Remote)
5 Months ago
Fortis Games - Lead Marketing Analyst

Fortis Games

Canada (On-Site)
5 Months ago
The Walt Disney Company - Technical Assistant

The Walt Disney Company

London, England, United Kingdom (Hybrid)
1 Month ago
Riot Games - Senior Principal Technical Artist

Riot Games

Dublin, County Dublin, Ireland (On-Site)
5 Months ago
Tencent - Machine Learning Platform Development Intern

Tencent

(On-Site)
1 Month ago
Canva - Senior Software Engineer -Cloud Platform- - Remote across ANZ

Canva

Sydney, New South Wales, Australia (Remote)
5 Months ago
The Walt Disney Company - Manager, Software Engineering – Media and Playback Data Processing

The Walt Disney Company

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

Get notifed when new similar jobs are uploaded

Jobs in Bengaluru, Karnataka, India

CleverTap - Manager - Product Design

CleverTap

Mumbai, Maharashtra, India (On-Site)
5 Months ago
CloudHire - Database Solution Architect

CloudHire

Mumbai, Maharashtra, India (Remote)
3 Weeks ago
Quizizz - Customer Success Manager

Quizizz

Bengaluru, Karnataka, India (Hybrid)
4 Months ago
EXUSIA - Power BI Data Analyst

EXUSIA

Hyderabad, Telangana, India (Remote)
1 Week ago
InvenioLSI - Sales Executive

InvenioLSI

New Delhi, Delhi, India (On-Site)
3 Months ago
CloudHire - Full Stack Developer - React and Node

CloudHire

Bengaluru, Karnataka, India (Remote)
2 Weeks ago
Nagarro - Associate Staff Engineer

Nagarro

Hyderabad, Telangana, India (On-Site)
6 Months ago
Microsoft - Senior Technical Program Manager

Microsoft

Hyderabad, Telangana, India (On-Site)
16 Hours ago
CloudHire - ML Engineer

CloudHire

Hyderabad, Telangana, India (Remote)
3 Weeks ago
Apollo Global Management,  Inc  - APPS Analyst

Apollo Global Management, Inc

Maharashtra, India (Hybrid)
7 Months ago

Get notifed when new similar jobs are uploaded

Research & Development Jobs

ByteDance - Software Engineer, AML Machine Learning Systems

ByteDance

Seattle, Washington, United States (On-Site)
2 Months ago
Microsoft - Research Intern - Networking

Microsoft

Redmond, Washington, United States (On-Site)
16 Hours ago
NVIDIA - Senior Firmware Engineer - Embedded Controller

NVIDIA

Santa Clara, California, United States (On-Site)
2 Days ago
ByteDance - Software Engineer, Model Interference

ByteDance

San Jose, California, United States (On-Site)
3 Months ago
Microsoft - Research Intern - AI HW/SW Co-design

Microsoft

Redmond, Washington, United States (On-Site)
16 Hours ago
Ethernovia - GUI Tools Software Development

Ethernovia

Pune, Maharashtra, India (Remote)
5 Months ago
Meta - Research Scientist Intern, Photorealistic Telepresence (PhD)

Meta

Pittsburgh, Pennsylvania, United States (On-Site)
5 Months ago
Google - Design Verification Engineer, ASIC

Google

Bengaluru, Karnataka, India (On-Site)
7 Hours ago
Dassault Systèmes - Software Engineer (Geometry)

Dassault Systèmes

Mumbai, Maharashtra, India (Hybrid)
5 Months ago
NVIDIA - Senior GPU Memory Architect

NVIDIA

Santa Clara, California, United States (On-Site)
3 Weeks 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.

Dublin, County Dublin, Ireland (On-Site)

Sunnyvale, California, United States (On-Site)

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

Warsaw, Masovian Voivodeship, Poland (On-Site)

Hyderabad, Telangana, India (On-Site)

Sunnyvale, California, United States (On-Site)

Sydney, New South Wales, Australia (On-Site)

Waterloo, Ontario, Canada (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