Staff Software Engineer, ML Hardware, YouTube Discovery

Google is a global technology company that runs the world's most popular video platform, YouTube.
$189,000 - $284,000
Machine Learning
Staff Software Engineer
In-Person
5,000+ Employees
8+ years of experience
AI

Description For Staff Software Engineer, ML Hardware, YouTube Discovery

YouTube, a part of Google, is seeking a Staff Software Engineer to lead their ML Hardware initiatives for the Discovery team. This role sits at the intersection of YouTube's recommendation systems and Google's cutting-edge ML hardware infrastructure. The position involves ensuring YouTube's business-critical recommender models optimally utilize available ML hardware, particularly the Tensor Processing Unit (TPU).

The role requires deep expertise in both software development and machine learning, with a focus on hardware acceleration and model optimization. You'll be responsible for shaping YouTube's ML hardware strategy, particularly in the context of growing model sizes and emerging technologies like Gemini and Large Language Models. This position offers an opportunity to impact how millions of users experience YouTube through improved recommendation systems.

The ideal candidate will bring 8+ years of software development experience, with specific expertise in ML infrastructure and large-scale ML models. Leadership experience is crucial, as you'll be coordinating with various teams across YouTube and Google. You'll work with state-of-the-art ML hardware and have the opportunity to influence future hardware developments.

This role offers competitive compensation, including a base salary range of $189,000-$284,000, plus bonus, equity, and comprehensive benefits. You'll be joining a company that values diversity, inclusion, and the belief that everyone deserves to have a voice. The position is based in San Bruno, CA, where you'll work with teams dedicated to pushing the boundaries of ML hardware optimization and recommendation systems.

Last updated 3 months 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
Vision Insurance
Dental Insurance
  • Base salary
  • Bonus
  • Equity
  • Medical Insurance
  • Vision Insurance
  • Dental Insurance

Interested in this job?

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

Senior Technical Program Manager II, Machine Learning, Google Cloud

Senior TPM role at Google Cloud leading machine learning initiatives, managing complex cross-functional projects, and driving strategic planning.

Senior Technical Program Manager I, Machine Learning, Google Cloud Platforms

Senior Technical Program Manager role at Google Cloud, focusing on Machine Learning initiatives with 8+ years of experience required.

Lead Group Product Manager, AI Frameworks

Lead Group Product Manager position at Google Cloud focusing on AI Frameworks, driving adoption and community building for technologies like JAX and PyTorch.

Senior Product Manager, On-Device Machine Learning, Developer Pillar

Senior Product Manager role at Google focusing on On-Device Machine Learning initiatives, requiring 8+ years of experience and offering comprehensive compensation package.

Senior Research Scientist, Google Cloud AI

Senior Research Scientist position at Google Cloud AI, focusing on advanced AI research and development with competitive compensation and benefits.