Search
Go
Staff Software Engineer, AI Engines 3P TPU Inference
MOUNTAIN VIEW · US
—
Full-time
STAFF
About the Role
Google is seeking a Staff Software Engineer to join the AI Engines team, focusing on third-party TPU inference. The role involves managing ML infrastructure, working with ML compilers and runtimes, and migrating popular frameworks such as TensorFlow, JAX, and PyTorch. Strong leadership and deep expertise in ML runtimes are essential.
Responsibilities
- Manage ML infrastructure for AI Engines
- Work on ML compilers and runtimes
- Migrate frameworks like TensorFlow, JAX, and PyTorch
Requirements
- Strong leadership skills
- Expertise in ML runtimes
- Experience with ML compilers and frameworks (TensorFlow, JAX, PyTorch)
Tech Stack
PyTorchTensorFlowJAX
Mountain View · US · 194668+ employees
Google is a multinational technology corporation specializing in internet-related services and products, including search engine technology, online advertising, cloud computing, AI, and consumer electronics. It is a major subsidiary of Alphabet Inc. and offers widely used platforms such as Gmail, Maps, Android, and YouTube.
