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 requires a mix of technical expertise, innovation, and collaboration to ensure that users have the best and fastest experience possible.

Key responsibilities include developing SoC architecture specifications, evaluating different silicon solutions for Google's data center AI accelerator roadmap, collaborating with Systems and Software teams for high-performance interfaces, and working with various stakeholders to ensure designs are complete, correct, and performant. The ideal candidate will have extensive experience in AI/ML SoC architecture, multi-chiplet SoC architectures, and ASIC design flows, along with proficiency in programming languages and knowledge of high-performance, low-power design techniques.