Machine Learning Hardware Architect, AI Infrastructure
Google · Sunnyvale, CA
About this role
Google is hiring a mid-level AI Infrastructure Engineer in the machine learning function based in Sunnyvale, CA. The posting calls out experience with GCP, Machine Learning, Performance Optimization, Vertex AI. Compensation is listed at $192,000–$278,000 per year.
- Role
- AI Infrastructure Engineer
- Function
- machine learning
- Level
- mid
- Track
- Individual contributor
- Employment
- Full-time
- Location
- Sunnyvale, CA
- Posted
- May 19, 2026
More roles at Google
Job description
from Google careersIn this role, you’ll work to shape the future of AI/ML hardware acceleration. You will have an opportunity to drive cutting-edge TPU (Tensor Processing Unit) technology that powers Google's most demanding AI/ML applications. You’ll be part of a team that pushes boundaries, developing custom silicon solutions that power the future of Google's TPU. You'll contribute to the innovation behind products loved by millions worldwide, and leverage your design and verification expertise to verify complex digital designs, with a specific focus on TPU architecture and its integration within AI/ML-driven systems.
As a Machine Learning Hardware Architect, you will influence the evolution of high-performance intelligence for next-generation computing infrastructure. You will have an opportunity to drive technology that powers large-scale systems where high throughput, reliability, and efficiency are mission-critical. You will be part of a team that pushes boundaries, developing solutions that define the future of intelligent data centers and enterprise hardware. You will contribute to the innovation behind products that transform industries, leveraging your expertise in system-level integration to deploy complex, large-scale AI models across sophisticated hardware platforms.
The AI and Infrastructure team is redefining what’s possible. We empower Google customers with breakthrough capabilities and insights by delivering AI and Infrastructure at unparalleled scale, efficiency, reliability and velocity. Our customers include Googlers, Google Cloud customers, and billions of Google users worldwide.