TPU Silicon Architect, Machine Learning Accelerator

Google welcomes people with disabilities.

Minimum qualifications:

• Master's degree in Computer Science, Electrical Engineering, related field, or equivalent practical experience.

• 4 years of experience in computer or chip architecture, performance, or compiler.

• Experience with one or more general purpose programming languages, including C/C++ or Python.

Preferred qualifications:

• PhD in Computer Science or Electrical Engineering.

• Experience with domain-specific accelerators.

• Experience with distributed/parallel programming.

• Experience with hardware/software co-design for machine learning.

• Experience with simulator development and micro-architecture.

About the job

In this role, you will join a high-impact team to develop the next generation machine learning accelerator architecture for on-device machine learning applications. It will require balancing short and long-term priorities to ensure we deliver on near-term architecture goals while continuing to invest in the longer-term priorities.

Google's mission is to organize the world's information and make it universally accessible and useful. Our team combines the best of Google AI, Software, and Hardware to create radically helpful experiences. We research, design, and develop new technologies and hardware to make computing faster, seamless, and more powerful. We aim to make people's lives better through technology.

Responsibilities

• Define the next generation machine learning accelerator architecture and ISA for on-device machine learning applications.

• Explore various trade-offs of future architecture designs in terms of performance, power, energy, and area.

• Collaborate with compiler engineers to optimize compilation results and extend the compiler infrastructure. Initiate new feature modeling in the architecture simulator.

• Collaborate with researchers and application developers to enable the latest machine learning work and to optimize the performance.

• Work collaboratively with full stack software engineers to partition machine learning workload to selected compute engines to enhance user experiences.

Google is proud to be an equal opportunity workplace and is an affirmative action employer. We are committed to equal employment opportunity regardless of race, color, ancestry, religion, sex, national origin, sexual orientation, age, citizenship, marital status, disability, gender identity or Veteran status. We also consider qualified applicants regardless of criminal histories, consistent with legal requirements. See also Google's EEO Policy and EEO is the Law. If you have a disability or special need that requires accommodation, please let us know by completing our Accommodations for Applicants form.