The Senior Software Engineer, Machine Learning role is part of the RecML team within Core ML's Applied ML organization at Google. This team focuses on accelerating product innovations through machine learning for recommendations and user modeling. The position involves working on ML infrastructure and efficiency to support the training framework of foundation recommender models. The role requires collaboration with various product areas to create standardized, maintainable, and production-grade solutions.

• Bachelor’s degree or equivalent practical experience.
• 5 years of experience programming in Python or C++.
• 3 years of experience with ML infrastructure, including model deployment and evaluation.

• Architect and implement the transition from data-parallel to model-parallel training paradigms.
• Design and manage large-scale training runs across multi-pod environments.
• Research and integrate transformer model optimizations to reduce training time and resource consumption.
• Write and optimize low-level model code to maximize performance out of the hardware.
• Work cross-functionally with the Kernel optimization team to implement compiler-level optimizations.

• Employees at Google are often offered benefits like comprehensive health insurance, 401(k) matching, and flexible work arrangements, among other benefits.