Taro Logo

Machine Learning Hardware Architect, Google Cloud

A global technology company that designs, implements, and manages hardware, software, machine learning, and systems infrastructure for Google services and Google Cloud.
$156,000 - $229,000
Machine Learning
Staff Software Engineer
In-Person
5,000+ Employees
5+ years of experience
AI · Enterprise SaaS · Cloud

Description For Machine Learning Hardware Architect, Google Cloud

Google Cloud is seeking a Machine Learning Hardware Architect to shape the future of AI/ML hardware acceleration, specifically focusing on TPU (Tensor Processing Unit) technology. This role sits at the intersection of hardware architecture and machine learning, requiring deep expertise in both domains.

The position is part of the ML, Systems, & Cloud AI (MSCA) organization, which is responsible for the hardware, software, and infrastructure powering Google's services and Cloud offerings. You'll be working on cutting-edge technology that powers Google's most demanding AI/ML applications, including the development of custom silicon solutions for TPUs.

Key responsibilities include creating architectural innovations for Google's TPU roadmap, evaluating system performance metrics, and collaborating across hardware, software, and ML teams. You'll be involved in workload characterization, benchmarking, and developing next-generation TPU features. The role requires a unique blend of hardware architecture knowledge and software development skills, particularly in C++ and Python.

The position offers competitive compensation ($156,000-$229,000 base salary) plus bonus, equity, and benefits. You'll be working in Sunnyvale, CA, alongside teams responsible for Google Cloud's Vertex AI and other leading AI platforms. This is an opportunity to impact the future of hyperscale computing and AI acceleration at one of the world's leading technology companies.

The ideal candidate will have at least 5 years of experience in computer architecture or related fields, with preferred qualifications including an advanced degree and experience with deep learning frameworks like TensorFlow and PyTorch. You'll be working on projects that directly influence Google's AI infrastructure and contribute to products used by millions worldwide.

This role offers the chance to work at the cutting edge of AI hardware development, combining technical expertise with practical implementation in a fast-paced, innovative environment. You'll be part of a team that's pushing the boundaries of what's possible in AI acceleration while maintaining Google's high standards for security, efficiency, and reliability.

Last updated 3 hours ago

Responsibilities For Machine Learning Hardware Architect, Google Cloud

  • Create differentiated architectural innovations for Google's semiconductor Tensor Processing Unit (TPU) roadmap
  • Evaluate the power, performance, and cost of prospective architecture and subsystems
  • Collaborate with partners in Hardware Design, Software, Compiler, ML Model and Research teams for hardware/software co-design
  • Work on Machine Learning (ML) workload characterization and benchmarking
  • Develop architecture for differentiating features on next generation TPUs

Requirements For Machine Learning Hardware Architect, Google Cloud

Python
  • Bachelor's degree in Electrical Engineering, Computer Engineering, Computer Science, a related field, or equivalent practical experience
  • 5 years of experience in computer architecture, chip architecture, IP architecture, co-design, performance analysis, or hardware design
  • Experience in developing software systems in C++ or Python

Benefits For Machine Learning Hardware Architect, Google Cloud

Medical Insurance
Equity
401k
  • Medical Insurance
  • Equity
  • 401k

Interested in this job?

Jobs Related To Google Machine Learning Hardware Architect, Google Cloud