The Performance Co-Design Engineer role at Google Cloud TPU focuses on shaping the future of AI and ML hardware acceleration. This position involves driving the development of custom silicon solutions that enhance Google's TPU technology, which powers demanding AI/ML applications. The engineer will collaborate with teams across AI research and engineering to innovate model architectures and optimize hardware performance. This role is crucial for integrating machine learning research with advanced silicon architectures to deliver high-performance accelerators.

• Bachelor's degree in Electrical Engineering, Computer Engineering, Computer Science, or a related field, or equivalent practical experience.
• 10 years of experience in computer architecture, chip architecture, or hardware-software co-design.
• Experience developing systems for performance modeling, simulation, or system analysis.
• Master's degree or PhD in Electrical Engineering, Computer Engineering or Computer Science, with an emphasis on computer architecture is preferred.
• Experience with deep learning frameworks such as TensorFlow or PyTorch.

• Drive the definition and optimization of the hardware/software stack to enable performant training and serving of large ML models.
• Collaborate with research and modeling teams to innovate on model architectures, focusing on scaling, quality, and their direct impact on hardware performance.
• Lead the development of configurable architectural simulators and cycle-accurate performance models to quantify microarchitectural optimizations.
• Conduct system-level performance analysis across highly distributed ML systems, innovating new methodologies to balance compute, memory bandwidth, and inter-chip network requirements.
• Engage with partners across hardware design, compiler development, and ML research to transition architectural innovations from concept to production.

• Employees at Google are often offered benefits like comprehensive health insurance, 401(k) matching, and flexible work arrangements, among other benefits.