Senior ML Systems Architect – TPU & AI Infra
Job in
Sunnyvale, Santa Clara County, California, 94087, USA
Listed on 2026-07-30
Listing for:
Socket.dev
Full Time
position Listed on 2026-07-30
Job specializations:
-
Software Development
Machine Learning/ ML Engineer, AI Engineer (Applied/Software)
Job Description & How to Apply Below
Google's Core ML and AI Infrastructure teams advance large-scale ML across Google Cloud TPUs, enabling massive training and inference with PyTorch and JAX. We seek engineers to shape ML frameworks, compilers, and hardware-aware software across Google products.
Engineers collaborate across teams, own critical ML infrastructure projects, and contribute to performance improvements on TPUs, Vertex AI, and data centers, with opportunities to switch teams as the company grows.
#J-18808-LjbffrPosition Requirements
10+ Years
work experience
To View & Apply for jobs on this site that accept applications from your location or country, tap the button below to make a Search.
(If this job is in fact in your jurisdiction, then you may be using a Proxy or VPN to access this site, and to progress further, you should change your connectivity to another mobile device or PC).
(If this job is in fact in your jurisdiction, then you may be using a Proxy or VPN to access this site, and to progress further, you should change your connectivity to another mobile device or PC).
Search for further Jobs Here:
×