Silicon AI/ML Architect, TPU

1 Month 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

Google - Software Engineer

Google

Sydney, New South Wales, Australia (On-Site)
1 Month ago
Google - Software Engineering Manager II

Google

Mountain View, California, United States (On-Site)
1 Month ago
Coda - Senior/Staff Software Engineer (Relocation to Bangkok)

Coda

Bratislava, Bratislava Region, Slovakia (Hybrid)
1 Year ago
Trend Micro - Large Language Models (LLM) Expert (VicOne_Automotive Security)

Trend Micro

Taipei City, Taiwan (On-Site)
8 Months ago
ION - Data Engineer/Analyst - Graduate Development Program

ION

Pisa, Tuscany, Italy (On-Site)
7 Months ago
Tesla - Student Software Developer - Vision/Image Processing

Tesla

Prüm, Rhineland-Palatinate, Germany (On-Site)
3 Months ago
Google - Senior System Engineer, Product Software

Google

New Taipei, New Taipei City, Taiwan (On-Site)
1 Month ago
Netflix - ML Software Engineer (L4/L5) - Media Algorithms

Netflix

Los Angeles, California, United States (On-Site)
1 Month ago
Google - CPU Design Verification Engineer

Google

Tel Aviv-Yafo, Tel Aviv District, Israel (On-Site)
1 Month ago
NXP - Software Internship – Automotive Security Firmware

NXP

Bucharest, Bucharest, Romania (On-Site)
8 Months ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Virtuos - Assistant Technical Director (Gameplay)

Virtuos

China (On-Site)
2 Months ago
ByteDance - AI/LLM Network Software Engineer (High Speed Network)

ByteDance

Seattle, Washington, United States (On-Site)
2 Months ago
ByteDance - Site Reliability Engineer Graduate (Product RD and Infrastructure-Global E-Commerce) - 2024 Start (BS/MS)

ByteDance

San Jose, California, United States (On-Site)
7 Months ago
Wargaming - Senior Game Designer, Meta (World of Tanks Blitz)

Wargaming

Vilnius, Vilnius County, Lithuania (Hybrid)
3 Months ago
GoDaddy - Senior Ceph Engineer

GoDaddy

(Remote)
1 Month ago
Casumo - AI Engineer

Casumo

(Hybrid)
2 Months ago
JustPlay - Head of Data

JustPlay

Berlin, Berlin, Germany (Hybrid)
2 Months ago
Mistplay - Senior DevOps Engineer II

Mistplay

Toronto, Ontario, Canada (Hybrid)
2 Months ago
ByteDance - Research Scientist, Code Generation

ByteDance

San Jose, California, United States (On-Site)
7 Months ago
Meta - Software Engineering Manager, Product Infrastructure

Meta

Washington, District Of Columbia, United States (Remote)
6 Months ago

Get notifed when new similar jobs are uploaded

Jobs in Bengaluru, Karnataka, India

Aryaka - Incident Manager

Aryaka

Bengaluru, Karnataka, India (On-Site)
3 Months ago
T systems - Java Front End Lead/ Architect

T systems

Bengaluru, Karnataka, India (Hybrid)
2 Months ago
ByteDance - Content Moderator - Nepali Speaker

ByteDance

Gurugram, Haryana, India (On-Site)
2 Months ago
Lakshya Digital - Senior 3D Art Lead - Character

Lakshya Digital

Haryana, India (On-Site)
2 Months ago
Vimeo - Financial Systems Manager

Vimeo

Bengaluru, Karnataka, India (On-Site)
1 Month ago
YouGov - Desktop Support Technician

YouGov

Navi Mumbai, Maharashtra, India (On-Site)
1 Month ago
Populla - Senior Software Engineer - Full Stack

Populla

Bengaluru, Karnataka, India (Hybrid)
1 Month ago
Xogar Games - VFX Artist Intern

Xogar Games

Bengaluru, Karnataka, India (On-Site)
1 Month ago
Magna International - ADAS Feature Owner

Magna International

Bengaluru, Karnataka, India (On-Site)
8 Months ago
Dentsu - APAC Data Architect & Engineer

Dentsu

Bengaluru, Karnataka, India (On-Site)
7 Months ago

Get notifed when new similar jobs are uploaded

Research & Development Jobs

ByteDance - Research Scientist in Generative AI for Science

ByteDance

San Jose, California, United States (On-Site)
2 Months ago
Krafton  - Global Live Service Operation Manager (inZOI)

Krafton

Seoul, South Korea (On-Site)
2 Months ago
GEA Group - HMI Software Engineer

GEA Group

Bengaluru, Karnataka, India (On-Site)
7 Months ago
NVIDIA - Physical Design Engineer

NVIDIA

Shanghai, Shanghai, China (On-Site)
4 Months ago
Netflix - Technical Program Manager (L5), Content Promotion & Distribution

Netflix

Los Angeles, California, United States (On-Site)
1 Month ago
Google - Senior Software Engineer, YouTube

Google

Bengaluru, Karnataka, India (On-Site)
1 Month ago
NVIDIA - Senior System Software Engineer, NCCL - Partner Enablement

NVIDIA

Austin, Texas, United States (Remote)
2 Months ago
Google - ASIC Engineer, IP Design, Silicon

Google

Mountain View, California, United States (On-Site)
1 Month ago
Google - Software Engineer, ML Compilers, Silicon

Google

Bengaluru, Karnataka, India (On-Site)
1 Month ago
NVIDIA - Senior Circuit Characterization Engineer

NVIDIA

Canada (Hybrid)
2 Months ago

Get notifed when new similar jobs are uploaded

About The Company

London, England, United Kingdom (On-Site)

Bengaluru, Karnataka, India (On-Site)

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

Bengaluru, Karnataka, India (On-Site)

Taipei City, Taiwan (On-Site)

Zürich, Zurich, Switzerland (On-Site)

Kirkland, Washington, United States (On-Site)

New Taipei, New Taipei City, Taiwan (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