Staff Software Engineer, AI/ML Compute

8 Hours ago • 8-11 Years • Artificial Intelligence • $197,000 PA - $291,000 PA

Job Summary

Job Description

Google seeks a Staff Software Engineer for AI/ML Compute. Responsibilities include partnering with customers to benchmark AI/ML model performance on Google Cloud, collaborating with internal infrastructure teams, conducting performance profiling and debugging, and designing specialized ML solutions. The role requires 8+ years of software development experience, 5+ years in product testing and launch, and 3+ years in ML infrastructure experience. Up to 10% travel may be required. The position involves working with demanding AI workloads and leveraging advanced ML infrastructure to enhance support for Google's services and Cloud customers.
Must have:
  • 8+ years software development
  • 5+ years product testing & launch
  • 3+ years ML infrastructure experience
  • Expertise in model deployment, evaluation, optimization
  • Proficient in cloud services (GCP preferred)
Good to have:
  • Master's/PhD in related field
  • Technical leadership experience
  • Experience in complex organizations
  • Direct customer collaboration
  • Data structures/algorithms expertise
Perks:
  • Bonus
  • Equity
  • Benefits

Job Details


Minimum qualifications:

  • Bachelor's degree or equivalent practical experience.
  • 8 years of experience in software development.
  • 5 years of experience testing, and launching software products.
  • 3 years of experience with software design and architecture.
  • 3 years of experience with ML infrastructure, including model deployment, evaluation, optimization, data processing, and debugging.

Preferred qualifications:

  • Master’s degree or PhD in Engineering, Computer Science, or a related technical field.
  • 8 years of experience with data structures/algorithms.
  • 3 years of experience in a technical leadership role leading project teams and setting technical direction.
  • 3 years of experience working in a complex, matrixed organization involving cross-functional, or cross-business projects.
  • Proven track record of collaborating directly with customers and field teams
  • Proficient in cloud services such as Compute, Storage, and Networking, preferably on GCP.

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.

The ML, Systems, & Cloud AI (MSCA) organization at Google designs, implements, and manages the hardware, software, machine learning, and systems infrastructure for all Google services (Search, YouTube, etc.) and Google Cloud. Our end users are Googlers, Cloud customers and the billions of people who use Google services around the world.

We prioritize security, efficiency, and reliability across everything we do - from developing our latest TPUs to running a global network, while driving towards shaping the future of hyperscale computing. Our global impact spans software and hardware, including Google Cloud’s Vertex AI, the leading AI platform for bringing Gemini models to enterprise customers.

The US base salary range for this full-time position is $197,000-$291,000 + bonus + equity + benefits. Our salary ranges are determined by role, level, and location. 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

  • Partner with customers to benchmark AI/ML model performance on Google Cloud compute and overall infrastructure.
  • Collaborate with internal infrastructure teams like Storage, Networking to enhance support for demanding AI workloads.
  • Conduct in-depth performance profiling, debugging, and troubleshooting of training and inference workloads.
  • Design and implement specialized ML solutions, effectively leveraging advanced ML infrastructure.
  • Role may include up to 10% business-related travel to the customers and partners within the US.

Similar Jobs

Fluxon - Senior Software Engineer

Fluxon

Bengaluru, Karnataka, India (Remote)
6 Months ago
Netflix - Manager, Performance Marketing

Netflix

Los Angeles, California, United States (On-Site)
16 Hours ago
Velotio Technologies - Software Engineer (ROR)

Velotio Technologies

Maharashtra, India (Remote)
2 Weeks ago
NVIDIA - Senior Software Engineer - Automated Parallel Programming

NVIDIA

Santa Clara, California, United States (Remote)
2 Months ago
NVIDIA - Senior ASIC Verification Engineer - GPU Memory Subsystem

NVIDIA

Santa Clara, California, United States (On-Site)
3 Weeks ago
Google - AI Sales Specialist III, ISV, West

Google

San Francisco, California, United States (On-Site)
11 Hours ago
Tencent - NLP Research Intern

Tencent

London, England, United Kingdom (On-Site)
5 Months ago
Google - Software Engineer III, AI/ML, Google Cloud Platforms

Google

Sunnyvale, California, United States (On-Site)
10 Hours ago
Dolby Laboratories - Senior Computer Vision Researcher

Dolby Laboratories

Bengaluru, Karnataka, India (Hybrid)
6 Months ago
ByteDance - Student Researcher (Doubao (Seed) - Foundation Model - MultiModal Generative Model)

ByteDance

San Jose, California, United States (On-Site)
2 Days ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Scopely - Software Engineer (Java)

Scopely

Bengaluru, Karnataka, India (Hybrid)
3 Months ago
ByteDance - Research Scientist in Large Model System

ByteDance

Seattle, Washington, United States (On-Site)
5 Months ago
NVIDIA - Machine Learning Intern - 2025

NVIDIA

(On-Site)
2 Months ago
ByteDance - Backend Software Engineer - Privacy & Security - Singapore

ByteDance

Singapore (On-Site)
5 Months ago
N-iX - Senior Engineer with AWS Greengrass Expertise

N-iX

Ukraine (Remote)
1 Month ago
Google - Software Engineer III, Site Reliability Engineering

Google

Warsaw, Masovian Voivodeship, Poland (On-Site)
8 Hours ago
Google - Software Engineer II, Migration and Replication, Google Cloud

Google

Tel Aviv-Yafo, Tel Aviv District, Israel (On-Site)
11 Hours ago
Canva - China App Store Marketing Partnerships Specialist

Canva

Beijing, Beijing, China (Remote)
4 Weeks ago
NVIDIA - Senior Synthesis Flow Development Engineer

NVIDIA

Santa Clara, California, United States (On-Site)
1 Month ago

Get notifed when new similar jobs are uploaded

Jobs in Kirkland, Washington, United States

Info Stretch - .Net Architect

Info Stretch

Mechanicsburg, Pennsylvania, United States (On-Site)
5 Months ago
ByteDance - Backend Software Engineer - Customer Service Platform - Seattle

ByteDance

Seattle, Washington, United States (On-Site)
5 Months ago
NVIDIA - Senior Manager, Hardware Engineering

NVIDIA

Santa Clara, California, United States (Hybrid)
2 Months ago
Google - Software Engineer III, Infrastructure, Google Cloud Compute Infrastructure

Google

Sunnyvale, California, United States (On-Site)
8 Hours ago
Patreon - iOS Engineer

Patreon

San Francisco, California, United States (Hybrid)
3 Weeks ago
Microsoft - Data Science PhD Internship

Microsoft

Mountain View, California, United States (On-Site)
19 Hours ago
ByteDance - University Relations Lead - Early Careers, AMS

ByteDance

New York, New York, United States (On-Site)
2 Days ago
PlayStation Global - Animator (External Development)

PlayStation Global

Los Angeles, California, United States (Hybrid)
3 Weeks ago
Passive Logic - AI Control Theory & Optimization Scientist

Passive Logic

Salt Lake City, Utah, United States (On-Site)
4 Months ago
Skybound Entertainment - Senior Manager, Brand Marketing

Skybound Entertainment

Los Angeles, California, United States (On-Site)
2 Months ago

Get notifed when new similar jobs are uploaded

Artificial Intelligence Jobs

Google - Senior Staff Software Engineer, Compilers, Research, Core ML

Google

Sunnyvale, California, United States (On-Site)
9 Hours ago
ByteDance - Student Researcher (Doubao (Seed) - Foundation Model - Generative AI)

ByteDance

Seattle, Washington, United States (On-Site)
3 Weeks ago
Meta - Software Engineer, Machine Learning

Meta

Sunnyvale, California, United States (On-Site)
5 Months ago
Scopely - Senior AI Creative (Motion) - Monopoly Go

Scopely

Barcelona, Catalonia, Spain (Hybrid)
4 Weeks ago
Google - Engineering Manager, GKE AI, Experience and Ecosystem

Google

Seattle, Washington, United States (On-Site)
9 Hours ago
Ubisoft - Senior ML Ops - Content Creation Technology Group

Ubisoft

Montreal, Quebec, Canada (On-Site)
3 Months ago
Match Group - Sr. Software Engineer, Generative AI

Match Group

Palo Alto, California, United States (Hybrid)
6 Months ago
Meta - Software Engineer, Systems ML - SW/HW Co-design

Meta

Seattle, Washington, United States (Remote)
5 Months ago
NVIDIA - Applied Research Intern - 2025

NVIDIA

Yerevan, Yerevan, Armenia (On-Site)
2 Months ago
SparkCognition - Data Scientist

SparkCognition

Bengaluru, Karnataka, India (On-Site)
6 Months 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.

Bucharest, Bucharest, Romania (On-Site)

Bengaluru, Karnataka, India (On-Site)

Bengaluru, Karnataka, India (On-Site)

Dublin, County Dublin, Ireland (On-Site)

Bengaluru, Karnataka, India (On-Site)

Sunnyvale, California, United States (On-Site)

Sunnyvale, 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