TPU Architect, Silicon

Google organizes the world's information and makes it universally accessible and useful, combining AI, Software, and Hardware to create helpful experiences.
New Taipei, Banqiao District, New Taipei City, Taiwan
Machine Learning
Mid-Level Software Engineer
In-Person
5,000+ Employees
3+ years of experience
AI · Hardware

Description For TPU Architect, Silicon

Google is seeking a TPU Architect to join their innovative hardware team focused on developing custom silicon solutions for Google's direct-to-consumer products. This role combines computer architecture expertise with machine learning acceleration, requiring deep understanding of both hardware and ML concepts. The position involves working with Google's Tensor Processing Units (TPUs), specialized hardware accelerators designed for machine learning workloads.

The ideal candidate will have strong foundations in computer architecture, with experience in microarchitecture, cache systems, and memory subsystems. They will work on analyzing and optimizing ML workloads, developing tools for performance analysis, and collaborating with implementation teams to enhance TPU architecture. Knowledge of ML algorithms and compiler optimization is highly valued.

This role offers the opportunity to shape the future of Google's AI hardware infrastructure, working at the intersection of machine learning and computer architecture. The position is based in New Taipei City, Taiwan, where you'll be part of a team pushing the boundaries of what's possible in ML acceleration hardware.

As a TPU Architect, you'll contribute to Google's mission of organizing the world's information by developing the hardware that powers their AI systems. You'll work with cross-functional teams, combining expertise in hardware architecture, machine learning, and performance optimization to create more efficient and powerful AI accelerators.

The role offers the chance to work on cutting-edge technology that impacts millions of users worldwide, with the backing of Google's resources and expertise in both hardware and AI. You'll be part of a team that values innovation, technical excellence, and collaborative problem-solving.

Last updated 7 hours ago

Responsibilities For TPU Architect, Silicon

  • 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

Requirements For TPU Architect, Silicon

Python
  • 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

Benefits For TPU Architect, Silicon

  • Equal Employment Opportunity
  • Accommodation for disabilities

Interested in this job?

Jobs Related To Google TPU Architect, Silicon

Field Solutions Developer II, Generative AI, Google Cloud

Field Solutions Developer II position at Google Cloud focusing on Generative AI implementation, requiring 5+ years of experience in AI/ML and Python programming.

Software Engineer III, Machine Learning, Search

Software Engineer III position at Google focusing on machine learning for Search, offering competitive compensation and the opportunity to work on large-scale AI systems.

Software Engineer III, Machine Learning, Google Ads

Software Engineer III position at Google focusing on machine learning applications in advertising technology, offering competitive compensation and opportunity to work on large-scale systems.

Software Engineer III, AI/ML, Google Cloud AI

Software Engineer III position at Google Cloud AI focusing on machine learning infrastructure and AI research, offering competitive compensation and opportunities to work on cutting-edge AI technologies.

Software Engineer III, Core Machine Learning, Google Cloud

Software Engineer III role at Google Cloud focusing on core machine learning, developing next-generation ML technologies and infrastructure.