Google is seeking software engineers for the Core ML team to advance ML workloads on TPU systems, including JAX and PyTorch, impacting Gemini and OSS models. You will contribute to performance optimization across large-scale ML projects and collaborate with researchers to push hardware and software boundaries.
Ideal candidates have extensive ML deployment experience, strong systems design skills, and a track record of delivering scalable, efficient software for massive-scale AI workloads.
#J-18808-LjbffrStaff ML Systems Engineer — TPU Performance & Gemini in sunnyvale at Unknown Company
This position is listed as full time and onsite.