Description
Google Core Machine Learning is hiring a Software Engineer to optimize Gemini and open-source machine learning models on TPU systems for JAX and PyTorch. The role focuses on ML training and serving benchmarks, identifying performance bottlenecks, improving compilers and runtimes, and enabling large-scale model training and inference across Google fleets. Candidates need a bachelor’s degree or equivalent practical experience, 8 years of software development experience, and substantial experience in ML infrastructure, design, testing, and software architecture. The position offers a US salary of $207,000–$300,000 plus a 20% bonus target, equity, and benefits, with interviews typically conducted in person in Sunnyvale or New York.
