Senior Software Engineer, Machine Learning, Core Ml

Google

Mountain View, CA, United States
Base salary only; bonus/equity: not specified; ben...
Not specified (assumed to be hybrid based on industry standards).
Model-parallel training paradigms
Multi-pod environment management
Transformer model optimization
Google's RecML team in the Core ML Applied ML organization is seeking a Senior Software Engineer specializing in machine learning to drive innovations in recommendations and user modeling. The role involves architecting and implementing advanced training paradigms and optimizing model performance across large-scale environments

Job Summary

  • The RecML team accelerates product innovations through applied research in recommendations and user modeling across Alphabet products.
  • This role involves architecting the transition from data-parallel to model-parallel training paradigms for foundation recommender models.
  • Engineers will write and optimize low-level model code including custom pallas kernels to maximize hardware performance.

Matching Summary

Match Score: 85

Google's RecML team in the Core ML Applied ML organization is seeking a Senior Software Engineer specializing in machine learning to drive innovations in recommendations and user modeling. The role involves architecting and implementing advanced training paradigms and optimizing model performance across large-scale environments.

Salary

Base salary only; Bonus/Equity: Not specified; Benefits: Not specified

Skills & Requirements

Must-have

  • Model-parallel training paradigms
  • Multi-pod environment management
  • Transformer model optimization
  • Custom pallas kernel development
  • Compiler-level optimization

Nice-to-have

  • Cross-functional collaboration skills
  • Applied research experience
  • Recommendation system expertise
  • User modeling background
  • Data center network bandwidth optimization

Key Requirements

  • Experience with large-scale training runs
  • Knowledge of multi-pod environments
  • Expertise in transformer architectures
  • Proficiency in low-level model coding
  • Understanding of compiler optimizations

Work Rights

Not specified

Tailored Resume

Cover Letter