Silicon AI/ML Architect, TPU, Google Cloud

Google is seeking a Silicon AI/ML Architect for their TPU team in Google Cloud. This role is part of a diverse team that pushes boundaries, developing custom silicon solutions that power the future of Google's direct-to-consumer products. As a Silicon AI/ML Architect, you will drive the Architecture of an AI/ML SoC and support engineering teams during implementation and validation phases. You'll work closely with internal system teams and external partners to develop a comprehensive understanding of required SoC functions. The role requires collaboration with Engineering teams to drive the optimal balance between performance, features, power, schedule, and cost. This position is crucial in shaping the next generation of hardware experiences, delivering unparalleled performance, efficiency, and integration for products loved by millions worldwide.

The Technical Infrastructure team at Google is responsible for building and maintaining the architecture that keeps everything running smoothly. From developing and maintaining data centers to building the next generation of Google platforms, this team makes Google's product portfolio possible. The role involves working with networks to ensure users have the best and fastest experience possible.

Key responsibilities include developing SoC architecture specifications, evaluating silicon solutions for Google's data center AI accelerator roadmap, collaborating with Systems and Software teams, and working closely with various stakeholders to ensure designs are complete, correct, and performant. The ideal candidate will have extensive experience in machine learning or graphics SoCs architecture, multi chiplet SoC architectures, and ASIC design flows.

Google is an equal opportunity employer committed to building a diverse and inclusive workforce. They offer accommodations for applicants with needs and require English proficiency for efficient global collaboration.