Senior ML Infra Engineer: PyTorch on Cloud TPUs

A leading tech company in California seeks a skilled ML framework developer to enhance PyTorch models running on Google Cloud TPUs. Responsibilities include collaborating with teams for innovation, supporting ML frameworks, and optimizing performance. The ideal candidate has strong experience in ML design, software development, and technical leadership, with a focus on AI capabilities. This position offers a competitive salary range of $166,000-$244,000 plus bonuses and benefits. #J-18808-Ljbffr