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, with models growing larger and requiring efficient training and serving using Google's 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 emerging technologies like 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 recommendation systems at one of the world's largest video platforms, while working with state-of-the-art ML hardware and infrastructure.

Last updated 4 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
Dental Insurance
Vision Insurance
401k
Equity
  • Comprehensive medical, dental, and vision insurance
  • 401k plan
  • Equity compensation
  • Bonus compensation

Interested in this job?

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

Staff Software Developer, Generative AI, Gemini Code Assist

Staff Software Developer position at Google focusing on Generative AI and Gemini Code Assist, requiring expertise in machine learning and software development.

Senior Research Scientist, NLP

Senior Research Scientist position at Google focusing on NLP and Large Language Models research and development in Tel Aviv.

Senior Research Scientist, Deep Learning Data

Senior Research Scientist position at Google focusing on Deep Learning Data research, graph algorithms, and Gemini Data infrastructure development.

Senior Research Scientist, Interactive Recommender Systems

Senior Research Scientist position at Google Research focusing on interactive recommender systems, machine learning, and AI, offering competitive compensation and research opportunities.

Senior Research Scientist, Multilingual NLP

Senior Research Scientist position at Google focusing on multilingual NLP and LLMs, requiring PhD and 7+ years of experience in machine learning and natural language processing.