Software Engineer III, Cloud TPU Compiler

Responsibilities

You will develop AI frameworks to enable PyTorch models to run efficiently on Google Cloud's TPUs and GPUs. Additionally, you will collaborate with researchers to maintain and optimize software for large-scale machine learning workloads.

Requirements

Candidates must have a bachelor's degree and at least two years of software development experience, including proficiency in C++. Experience with ML infrastructure and compilers is required, with advanced degrees and specific framework knowledge preferred.

Skills

• Software Engineering
• C++
• Machine Learning
• Cloud TPU
• PyTorch
• XLA
• Pallas
• GPU Programming
• Distributed Computing
• Compiler Development
• Model Deployment
• Data Processing
• Debugging
• System Design
• Artificial Intelligence
• Natural Language Processing

About this role

MINIMUM QUALIFICATIONS:

* Bachelor’s degree or equivalent practical experience.
* 2 years of experience with software development in one or more programming
languages, or 1 year of experience with an advanced degree.
* 1 year of experience with ML infrastructure (e.g., model deployment, model
evaluation, data processing, debugging).
* Experience in C++.

PREFERRED QUALIFICATIONS:

* Master's degree or PhD in Computer Science or related technical fields.
* Experience with ML frameworks, such as PyTorch/XLA/Pallas/GPU/TPU
programming.
* Experience with compilers, machine learning optimization, distributed machine
learning.

ABOUT THE JOB:

Google's software engineers develop the next-generation technologies that change
how billions of users connect, explore, and interact with information and one
another. Our products need to handle information at massive scale, and extend
well beyond web search. We're looking for engineers who bring fresh ideas from
all areas, including information retrieval, distributed computing, large-scale
system design, networking and data storage, security, artificial intelligence,
natural language processing, UI design and mobile; the list goes on and is
growing every day. As a software engineer, you will work on a specific project
critical to Google’s needs with opportunities to switch teams and projects as
you and our fast-paced business grow and evolve. We need our engineers to be
versatile, display leadership qualities and be enthusiastic to take on new
problems across the full-stack as we continue to push technology forward.

Google’s revolutionary Cloud TPUs were designed from the ground up to accelerate
machine learning workloads, providing the computational power to train and run
cutting-edge machine learning models in a fraction of the time and cost
conventionally required. TPUs have been wildly successful internally at Google.
With our Cloud TPU service, it's easy to take your existing ML workloads
(PyTorch, JAX, etc.) and move them to the cloud to take advantage of this great
capability.

Google Cloud accelerates every organization’s ability to digitally transform its
business and industry. We deliver enterprise-grade solutions that leverage
Google’s cutting-edge technology, and tools that help developers build more
sustainably. Customers in more than 200 countries and territories turn to Google
Cloud as their trusted partner to enable growth and solve their most critical
business problems.

RESPONSIBILITIES:

* Work on AI framework development to successfully enable PyTorch models to run
on Google Cloud's TPUs and GPUs and tune for peak performance.
* Provide comprehensive support for ML frameworks and compilers on Cloud TPUs
and GPUs, enabling the training and deployment of the most advanced machine
learning models, driving innovation and breakthroughs.
* Collaborate with other Google teams and leading researchers across the
industry to continuously bring cutting-edge ML capabilities to our
PyTorch-in-Cloud offering.
* Help with design, development, testing deploying and maintaining software.

About Google

Software Development · 10,001+ employees · headquartered in Mountain View, CA