Search
Go
Staff Software Engineer, AI Engines 3P TPU Inference
Summary by AIEngineer.careers
Staff Software Engineer, AI Engines 3P TPU Inference is a AI Infrastructure Engineer role (full-time). with Google. in MOUNTAIN VIEW, US. Imported listing (source: localjobs.com). Apply on the employer's site (localjobs.com).
Imported listing (source: localjobs.com) · Apply on localjobs.com
MOUNTAIN VIEW · US
—
Full-time
STAFF
The sections below reproduce the third-party job description for reference. AIEngineer.careers does not write or control this text.
Imported job description
Sourced from localjobs.com
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
JAXPyTorchTensorFlow
Mountain View · US · 190820+ employees
Google is a multinational technology corporation specializing in internet-related services, including search, cloud computing, software, consumer electronics, and artificial intelligence. It serves as a subsidiary of Alphabet Inc. and provides a wide range of web-based products for consumers and enterprises.
