Taro Logo

Senior Software Engineer, ChromeOS, On-Device Machine Learning

Google develops next-generation technologies that change how billions of users connect, explore, and interact with information and one another.
Machine Learning
Senior Software Engineer
In-Person
5,000+ Employees
5+ years of experience
AI

Job Description

Google is seeking a Senior Software Engineer to join their ChromeOS On-Device Machine Learning team. This role focuses on developing and optimizing machine learning and generative AI models for ChromeOS devices. The position combines deep technical expertise in ML with practical implementation skills, working on model quality, compilation, and API development for user experiences.

The role involves working with various compute platforms (CPU, GPU, NPUs) across laptops and tablets, requiring expertise in ML modeling, neural networks, transformers, and GenAI. The successful candidate will be responsible for optimizing model performance, implementing fine-tuning techniques, and ensuring efficient model deployment across Chrome OS devices.

ChromeOS's mission is to provide universal access to computing through fast, secure, and intelligent solutions. This position offers the opportunity to work on cutting-edge ML technologies while impacting billions of users. The team collaborates across various departments to optimize power usage, compute efficiency, and memory footprint of ML models.

The ideal candidate should have strong experience in software development, ML infrastructure, and performance optimization. Knowledge of ML frameworks, on-device deployment, and hardware-accelerated ML inference techniques is highly valued. This role offers the chance to drive innovation in on-device ML while working with industry-leading technologies and frameworks.

Last updated 6 days ago

Responsibilities For Senior Software Engineer, ChromeOS, On-Device Machine Learning

  • Bringup ML and GenAI models onto various compute (CPU, GPU and NPUs) across suite of devices
  • Test and benchmark model performance and quality across varying sizes and constraints
  • Fine-tune training and model quality optimizations. Also model compilation and training
  • Build inference graphs that can leverage on-device models
  • Collaborate with power and performance teams to optimize model power/compute usage and memory footprint

Requirements For Senior Software Engineer, ChromeOS, On-Device Machine Learning

Python
  • Bachelor's degree or equivalent practical experience
  • 5 years of experience with software development in one or more programming languages (e.g., Python, C, C++)
  • 5 years of experience testing, and launching software products
  • 3 years of experience leading technical project strategy, ML design, and working with industry-scale ML infrastructure
  • 3 years of experience in performance analysis and optimization including GPU programming