Taro Logo

Silicon AI/ML Architect, TPU, Google Cloud

Google is a global technology company that develops innovative products and services used by millions worldwide.
Machine Learning
Principal Software Engineer
In-Person
5,000+ Employees
10+ years of experience
AI · Enterprise SaaS
This job posting may no longer be active. You may be interested in these related jobs instead:

Description For Silicon AI/ML Architect, TPU, Google Cloud

Google is seeking a Silicon AI/ML Architect to join their Technical Infrastructure team, focusing on TPU (Tensor Processing Unit) development for Google Cloud. This role combines hardware architecture expertise with machine learning specialization, working on custom silicon solutions that power Google's future products. The position requires deep understanding of SoC architecture, AI/ML systems, and hardware-software integration.

As a Silicon AI/ML Architect, you'll be at the forefront of developing cutting-edge AI accelerator architectures, working with multi-chiplet SoC designs and optimizing for performance, power, and area efficiency. You'll collaborate with various teams across Google, from systems and software engineers to verification and silicon validation experts, ensuring the delivery of high-performance AI/ML hardware solutions.

The role offers the opportunity to shape the future of Google's AI infrastructure, working on projects that directly impact millions of users worldwide. You'll be part of the team that develops and maintains Google's data centers and builds next-generation platforms. The position requires a blend of technical expertise in hardware architecture, machine learning, and system design, with a focus on creating innovative solutions for AI acceleration.

This is an ideal role for someone who has extensive experience in SoC architecture, particularly in the AI/ML domain, and wants to work on challenging problems at scale. You'll be contributing to Google's mission of advancing AI technology while working with some of the industry's best engineers and researchers. The role offers the chance to work on unique technical challenges while being part of a diverse and inclusive team that pushes boundaries in hardware innovation.

Last updated 7 months ago

Responsibilities For Silicon AI/ML Architect, TPU, Google Cloud

  • 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
  • Collaborate closely with Systems and Software teams to create high performance hardware/software interfaces
  • Collaborate closely with design, verification, emulation, physical design, packaging, and silicon validation stakeholders
  • Evaluate power, performance, area tradeoffs and drive improvements across generations

Requirements For Silicon AI/ML Architect, TPU, Google Cloud

Python
Go
  • 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
  • Experience in SoC designs, RTL coding and integration flows
  • Proficiency in a procedural programming language (e.g., C++, Python, Go)
  • Knowledge of arithmetic units, bus architectures, accelerators, or memory hierarchies
  • Knowledge of high performance and low power design techniques

Benefits For Silicon AI/ML Architect, TPU, Google Cloud

Medical Insurance
Vision Insurance
Dental Insurance
Parental Leave
  • Equal employment opportunity
  • Inclusive work environment
  • Comprehensive medical, dental, and vision insurance
  • Parental leave benefits
  • Accommodation for special needs

Interested in this job?