Staff Software Engineer, ML Hardware, YouTube Discovery

Google is a global technology leader that operates YouTube, the world's largest video-sharing platform.
$189,000 - $284,000
Machine Learning
Staff Software Engineer
In-Person
5,000+ Employees
8+ years of experience
AI · Enterprise SaaS

Description For Staff Software Engineer, ML Hardware, YouTube Discovery

YouTube, a division of Google, is seeking a Staff Software Engineer to lead their ML Hardware initiatives for the Discovery team. This role is crucial as YouTube's growth is powered by Machine Learning recommendations, which require efficient training and serving using Google's ML hardware Tensor Processing Unit (TPU). The position involves managing YouTube's participation in hardware development and evaluation programs, while also driving efforts to adapt YouTube's models for new accelerator capabilities.

The ideal candidate will be at the forefront of implementing cutting-edge ML hardware solutions, working with large-scale recommender systems, and Gemini and Large Language Models. They will be responsible for optimizing YouTube's business-critical recommender models to best utilize available ML hardware, including initiatives in quantized training and inference.

Working at YouTube means being part of a culture that believes in giving everyone a voice and making the world better through shared stories. The team operates at the intersection of cutting-edge technology and creativity, moving at the speed of culture with the goal of showing people the world. The collaborative environment encourages exploration of new ideas, solving real problems, and having fun while doing it.

The position offers a competitive compensation package including a base salary range of $189,000-$284,000, plus bonus, equity, and comprehensive benefits. This is an opportunity to join a team that's shaping the future of content discovery and ML hardware optimization at one of the world's largest video platforms.

Last updated 10 days ago

Responsibilities For Staff Software Engineer, ML Hardware, YouTube Discovery

  • Develop YouTube Discovery's ML hardware adoption strategy
  • Initiate and lead engineering efforts to adapt YouTube's recommender models to perform efficiently on future generations of ML hardware
  • Lead YouTube's evaluation of new ML hardware, in collaboration with model developers and Google-wide ML hardware and software experts

Requirements For Staff Software Engineer, ML Hardware, YouTube Discovery

Python
  • Bachelor's degree or equivalent practical experience
  • 8 years of experience in software development, and with data structures/algorithms
  • 5 years of experience with ML infrastructure (e.g., model deployment, model evaluation, optimization, debugging)
  • 3 years of experience in developing large-scale ML models utilizing ML hardware accelerators
  • 2 years of experience in a technical leadership role
  • Experience with post-training quantization, quantized aware training, or quantized training for ML models
  • Experience with building efficient and reusable AI infrastructure, compilers, or performance engineering
  • Experience with optimizing ML models to efficiently run on ML hardware accelerators
  • Excellent communication skills

Benefits For Staff Software Engineer, ML Hardware, YouTube Discovery

Medical Insurance
Equity
  • Base salary
  • Bonus
  • Equity
  • Comprehensive benefits package

Interested in this job?

Jobs Related To Google Staff Software Engineer, ML Hardware, YouTube Discovery

Senior Staff Research Scientist

Senior Staff Research Scientist position at Google Research, focusing on machine learning and AI development, offering competitive salary and benefits in Cambridge, MA.

Staff Research Scientist, Machine Learning Efficiency

Lead Machine Learning efficiency research at Google, focusing on next-generation language models and TPU development.

Senior Research Scientist, NLP

Senior Research Scientist position at Google focusing on NLP research and Large Language Models, requiring PhD and research experience.

Senior Research Scientist

Senior Research Scientist position at Google Research, focusing on machine learning and AI development with competitive compensation and benefits.

Senior Staff Research Scientist

Senior Staff Research Scientist position at Google Research, focusing on machine learning and AI development with opportunities for publication and research leadership.