Staff Software Engineer, ML Hardware, YouTube Discovery
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.
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
- 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
- Base salary
- Bonus
- Equity
- Comprehensive benefits package
Jobs Related To Google Staff Software Engineer, ML Hardware, YouTube Discovery
