hero

Explore Career Opportunities with GSV’s world-class partners

Powered By

Nu Advisory Partners
28
companies
802
Jobs

TPU Architect, Machine Learning Accelerator

Google

Google

Software Engineering, IT
South Korea · Seoul, South Korea
Posted on Thursday, May 25, 2023

Google welcomes people with disabilities.

Qualifications

Minimum qualifications:

  • Master's degree in Computer Science, Electrical Engineering, related field, or equivalent practical experience.
  • 6 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 optimize the performance.
  • Work collaboratively with full stack software engineers to partition machine learning workload to selected compute engines to enhance user experiences.

Information collected and processed as part of your Google Careers profile, and any job applications you choose to submit is subject to Google's Applicant and Candidate Privacy Policy.

To all recruitment agencies: Google does not accept agency resumes. Please do not forward resumes to our jobs alias, Google employees or any other organization location. Google is not responsible for any fees related to unsolicited resumes.

At Google, we’re committed to building a workforce that is more representative of the users we serve and creating a culture where everyone feels like they belong. To learn more about our diversity, equity, inclusion commitments and how we’re building belonging, please visit our Belonging page for more information.

We welcome and encourage people who are expecting and/or parents-to-be to apply to this or any other role at Google.

Google is a global company and, in order to facilitate efficient collaboration and communication globally, English proficiency is a requirement for all roles.