Staff Software Engineer, Platform-Aware AutoML

1 Month ago • 8-13 Years • Artificial Intelligence • $189,000 PA - $284,000 PA

Job Summary

Job Description

This Staff Software Engineer role focuses on developing Platform-Aware AutoML, a system that optimizes Machine Learning model performance and hardware utilization. Responsibilities include developing efficient job scheduling, researching and developing ML workloads for various Google products (Search, Ads, LLMs), understanding hardware architectures for automated optimization, optimizing critical ML workloads, and productionizing reinforcement learning algorithms. The role requires expertise in ML, software development, and system design, with a focus on maximizing AI/ML training efficiency.
Must have:
  • 8+ years software development experience
  • 5+ years ML design & infrastructure optimization
  • 5+ years testing & product launch experience
  • Python proficiency
  • Develop efficient ML job scheduling
  • Optimize business-critical ML workloads
Good to have:
  • Master's/PhD in relevant field
  • Technical leadership experience
  • Cross-functional project experience
  • Experience with various ML domains (Deep Learning, Reinforcement Learning, etc.)
  • Knowledge of computer architecture and performance analysis

Job Details

Minimum qualifications:

  • Bachelor’s degree or equivalent practical experience.
  • 8 years of experience in software development, and with data structures or algorithms.
  • 5 years of experience in testing, and launching software products, and 3 years of experience with software design and architecture.
  • 5 years of experience leading Machine Learning design and optimizing Machine Learning infrastructure (e.g., model deployment, model evaluation, data processing, debugging, fine tuning, etc.).
  • Experience developing software applications using Python.

Preferred qualifications:

  • Master’s degree or PhD in Engineering, Computer Science, a related technical field, or equivalent practical experience.
  • 3 years of experience in a technical leadership role and setting technical direction.
  • 3 years of experience in working cross-functionally, or cross-business projects.
  • Experience with Machine Learning (ML) based performance work.
  • Experience in spanning across various Machine Learning (ML) domains (e.g., Deep Learning, Reinforcement Learning, Neural Networks, Autoregressive Models, etc.).
  • Knowledge of computer architecture and performance analysis.

About the job

Google's software engineers develop the next-generation technologies that change how billions of users connect, explore, and interact with information and one another. Our products need to handle information at massive scale, and extend well beyond web search. We're looking for engineers who bring fresh ideas from all areas, including information retrieval, distributed computing, large-scale system design, networking and data storage, security, artificial intelligence, natural language processing, UI design and mobile; the list goes on and is growing every day. As a software engineer, you will work on a specific project critical to Google’s needs with opportunities to switch teams and projects as you and our fast-paced business grow and evolve. We need our engineers to be versatile, display leadership qualities and be enthusiastic to take on new problems across the full-stack as we continue to push technology forward.

Machine Learning models are becoming difficult to get the best performance out of the hardware they run on and use our dynamic computers as efficiently as possible. To solve this, you will be developing a system called Platform-Aware Auto Machine Learning (ML), which will automatically finds the best settings for both the model and the hardware, so we can get the most out of our machines, Platform-Aware AutoML helps researchers, computer architects, and people using Machine Learning work together to make the most of the latest technology.

In this role, you will be responsible for performance and extracting maximum for AI/ML training workloads. You will drive Machine Learning (ML) performance by identifying performance opportunities in Google production and research Machine Learning workloads, landing optimizations.

Google Cloud accelerates every organization’s ability to digitally transform its business and industry. We deliver enterprise-grade solutions that leverage Google’s cutting-edge technology, and tools that help developers build more sustainably. Customers in more than 200 countries and territories turn to Google Cloud as their trusted partner to enable growth and solve their most critical business problems.

The US base salary range for this full-time position is $189,000-$284,000 + bonus + equity + benefits. Our salary ranges are determined by role, level, and location. The range displayed on each job posting reflects the minimum and maximum target salaries for the position across all US locations. Within the range, individual pay is determined by work location and additional factors, including job-related skills, experience, and relevant education or training. Your recruiter can share more about the specific salary range for your preferred location during the hiring process.

Please note that the compensation details listed in US role postings reflect the base salary only, and do not include bonus, equity, or benefits. Learn more about .

Responsibilities

  • Develop Machine Learning job scheduling efficiency to match the best Machine Learning hardware with ML models.
  • Conduct research and development to gain understanding and enable business critical Machine Learning workloads (e.g., Search, Ads, LLM, etc.) with Platform-Aware AutoML.
  • Establish understanding of the latest Machine Learning platform hardware architecture to enable automated optimization.
  • Optimize business critical Machine Learning workloads to gain insights on how these optimizations can be automated at scale.
  • Develop, and productionize effective reinforcement learning algorithms to Platform-Aware Auto Machine Learning.

Similar Jobs

Patterned Learning Career - Junior QA Assurance Analyst  - Remote Job, 1+ Year Experience

Patterned Learning Career

(Remote)
1 Month ago
Google - Software Engineer III, AI/ML

Google

(On-Site)
1 Month ago
Patterned Learning Career - Junior Frontend Engineer (JavaScript/HTML/CSS

Patterned Learning Career

(Remote)
1 Month ago
Playrix - Feature Owner (LiveOps)

Playrix

Georgia (Remote)
3 Months ago
Social Discovery Ventures - Product Owner, MarTech

Social Discovery Ventures

Spain (Remote)
4 Months ago
Unity - Principal Machine Learning Engineer

Unity

San Francisco, California, United States (On-Site)
2 Months ago
PristineAI - AI Engineer

PristineAI

Chennai, Tamil Nadu, India (On-Site)
7 Months ago
Google - Senior Engineering Manager, ML Performance

Google

(On-Site)
1 Month ago
Stylumia - Senior Machine Learning Engineer - Time Series & Computer Vision

Stylumia

Bengaluru, Karnataka, India (Hybrid)
5 Months ago
Terralogic - SOFTWARE ENGINEER – AIML QA

Terralogic

Bengaluru, Karnataka, India (On-Site)
6 Months ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

G5 Games - 2D Illustrator (Match-3 project)

G5 Games

Astana, Astana, Kazakhstan (Remote)
1 Month ago
Patterned Learning Career - Junior Data Engineer  - Remote Job, 1+ Year Experience

Patterned Learning Career

(Remote)
1 Month ago
Codeninja - AI Research Scientist

Codeninja

Sindh, Pakistan (On-Site)
1 Month ago
Patterned Learning Career - Junior Web development programmer

Patterned Learning Career

(Remote)
1 Month ago
G5 Games - 2D UI/UX Artist (match-3 project)

G5 Games

Sofia, Sofia City Province, Bulgaria (Remote)
3 Months ago
DISCO - Software Engineer III, Backend

DISCO

Gurugram, Haryana, India (On-Site)
4 Months ago
Patterned Learning Career - Junior Website Developer - Remote Job, 1+ Year Experience

Patterned Learning Career

(Remote)
1 Month ago
Patterned Learning Career - .Junior Data Scientist  - Remote Job, 1+ Year Experience

Patterned Learning Career

(Remote)
1 Month ago
Patterned Learning Career - Junior Java AWS Developer

Patterned Learning Career

(Remote)
1 Month ago
Google - AI/ML System Architect, Silicon

Google

(On-Site)
1 Month ago

Get notifed when new similar jobs are uploaded

Jobs in Sunnyvale, California, United States

Wolters Kluwer - Manager, Product Software Engineering | AI

Wolters Kluwer

Coppell, Texas, United States (Hybrid)
3 Months ago
Matic Robots - Hardware Reliability Engineer - Electrical

Matic Robots

Mountain View, California, United States (On-Site)
3 Months ago
ByteDance - XR Mechanical Engineer / Architect- Pico Lab- San Jose

ByteDance

San Jose, California, United States (On-Site)
3 Months ago
Kyruus Health - Senior Technical Project Manager

Kyruus Health

United States (On-Site)
3 Months ago
BrightEdge - Sales Development Representative (BGSU Students)

BrightEdge

Cleveland, Ohio, United States (On-Site)
3 Months ago
Modulate - Senior Machine Learning Engineer

Modulate

Somerville, Massachusetts, United States (Hybrid)
1 Month ago
Magnopus - Previs Lead

Magnopus

Los Angeles, California, United States (Hybrid)
4 Months ago
CAE - Senior Manager, TPD Operations

CAE

Tampa, Florida, United States (On-Site)
3 Months ago
Mass Virtual - Technical Level Designer

Mass Virtual

Orlando, Florida, United States (On-Site)
5 Months ago
Next Level Business Services - Teradata Developer

Next Level Business Services

St. Louis, Missouri, United States (On-Site)
3 Months ago

Get notifed when new similar jobs are uploaded

Artificial Intelligence Jobs

Blizzard Entertainment - Senior Manager, Data Science

Blizzard Entertainment

United States (Hybrid)
3 Months ago
Dolby Laboratories - AIOps Research Scientist

Dolby Laboratories

Bengaluru, Karnataka, India (Hybrid)
3 Months ago
The Walt Disney Company - Lead Machine Learning Engineer, Ad Platforms

The Walt Disney Company

Seattle, Washington, United States (On-Site)
2 Months ago
Google - Research Intern, PhD, Summer 2025

Google

Atlanta, Georgia, United States (On-Site)
2 Months ago
PlayStation Global - Staff Machine Learning Engineer

PlayStation Global

London, England, United Kingdom (Hybrid)
3 Months ago
Microsoft - Principal Software Engineer (Infra)

Microsoft

Vancouver, British Columbia, Canada (On-Site)
1 Month ago
Google - Staff Software Engineer, Machine Learning, Search

Google

Mountain View, California, United States (On-Site)
1 Month ago
Ubisoft - Scientifique principal en données ML _ Groupe Technologique Content Creation

Ubisoft

Montreal, Quebec, Canada (On-Site)
1 Month ago

Get notifed when new similar jobs are uploaded

About The Company

A problem isn't truly solved until it's solved for all. Googlers build products that help create opportunities for everyone, whether down the street or across the globe. Bring your insight, imagination and a healthy disregard for the impossible. Bring everything that makes you unique. Together, we can build for everyone.

Hyderabad, Telangana, India (On-Site)

New Taipei, New Taipei City, Taiwan (On-Site)

New York, New York, United States (On-Site)

Dublin, County Dublin, Ireland (On-Site)

Kuala Lumpur, Federal Territory Of Kuala Lumpur, Malaysia (On-Site)

Mountain View, California, United States (On-Site)

View All Jobs

Get notified when new jobs are added by Google

Level Up Your Career in Game Development!

Transform Your Passion into Profession with Our Comprehensive Courses for Aspiring Game Developers.

Job Common Plug