Senior AI/ML Software Engineer — HPC & Large-Scale Systems

Google

United States

On-site

USD 207,000 - 301,000

Full time

14 days+

Get more replies from employers

Send a job-specific resume in minutes.

Job summary

Google Public Sector is seeking a Staff Software Engineer, AI/ML, to lead advanced AI/ML initiatives in Reston, VA. You will architect large-scale distributed data systems, optimize ML inference on specialized accelerators, and guide cross-functional teams across the product lifecycle.

This role requires deep expertise in software engineering, 8 years of experience, and strong leadership. Familiarity with Docker/Kubernetes and cloud storage is essential; security-clearance eligibility is a plus.

Qualifications

  • Bachelor's degree or equivalent practical experience.
  • 8 years of experience programming in C++, Java, Python, Kotlin, or Go.
  • Experience in technical leadership, roadmaps, and delivering projects.
  • Experience in parallel computing, hardware-level optimization, and accelerator optimization.
  • Master's or PhD in CS/Physics/Math preferred.
  • 8 years building large-scale distributed data systems and ML deployments.
  • Experience deploying DL architectures with PyTorch or TensorFlow on large clusters.
  • Experience with Docker, Kubernetes, and distributed storage.

Responsibilities

  • Architect and operate data synthesis pipelines and AI-based retrieval apps.
  • Manage petabyte-scale data ingestion across compute environments.
  • Optimize ML inference for accelerators.
  • Lead technical direction and mentor engineering teams.
  • Implement data lifecycle policies for resilience and integrity.

Skills

C++/Java/Python
Kotlin/Go
Technical leadership
Parallel computing
Low-level optimization
HPC

Education

Bachelor's degree or equivalent

Tools

Docker
Kubernetes
Distributed filesystems

Job description

Google Public Sector is seeking a Staff Software Engineer, AI/ML, to lead advanced AI/ML initiatives in Reston, VA. You will architect large-scale distributed data systems, optimize ML inference on specialized accelerators, and guide cross-functional teams across the product lifecycle.

This role requires deep expertise in software engineering, 8 years of experience, and strong leadership. Familiarity with Docker/Kubernetes and cloud storage is essential; security-clearance eligibility is a plus.

Get your free, confidential resume review.
or drag and drop your file here.
Similar jobs

Similar jobs worth comparing

Senior AI/ML Software Engineer — HPC & Distributed ML
Senior AI/ML Software Engineer — HPC & Distributed ML

Google • Reston (VA)

On-site
USD 207,000 - 301,000
Staff AI/ML Engineer — Large-Scale Data & HPC Systems
Staff AI/ML Engineer — Large-Scale Data & HPC Systems

Google Inc. • Reston (VA)

On-site
USD 207,000 - 300,000
Bonus
Equity
Benefits
Staff Software Engineer, AI/ML, Google Public Sector
Staff Software Engineer, AI/ML, Google Public Sector

Google Inc. • Reston (VA)

On-site
USD 207,000 - 300,000
Bonus
Equity
Benefits
Staff AI/ML Engineer — Cloud Security & Infra
Staff AI/ML Engineer — Cloud Security & Infra

Google • United States

On-site
USD 207,000 - 301,000
Senior AI/ML Software Engineer — Equity & Large‑Scale Infra
Senior AI/ML Software Engineer — Equity & Large‑Scale Infra

Google • Mountain View (CA)

On-site
USD 174,000 - 253,000
Staff AI/ML Storage Infrastructure Engineer
Staff AI/ML Storage Infrastructure Engineer

Google Inc. • San Jose (CA)

On-site
USD 207,000 - 301,000
20% bonus target
Equity
Comprehensive benefits
Staff Software Engineer, AI/ML, Google Public Sector
Staff Software Engineer, AI/ML, Google Public Sector

Google • Reston (VA)

On-site
USD 207,000 - 301,000
Senior Staff ML Systems Engineer - Lead AI & Scale
Senior Staff ML Systems Engineer - Lead AI & Scale

Google • Kirkland (WA)

On-site
USD 207,000 - 301,000
Health insurance
Dental insurance
Vision insurance
+8
Staff Software Engineer, AI/ML, Google Public Sector
Staff Software Engineer, AI/ML, Google Public Sector

Google • United States

On-site
USD 207,000 - 301,000
Senior Staff AI/ML GenAI Engineer, Scalable Cloud AI
Senior Staff AI/ML GenAI Engineer, Scalable Cloud AI

Google Inc. • New York (NY)

On-site
USD 262,000 - 365,000
Equity
Bonus target (25%)
Benefits