Staff Software Engineer, AI/ML, Google Public Sector
Back to JobsGoogle Get Smart Job AI Coach in the appFree on iOS and Android 
Google
Google


Google
Staff Software Engineer, AI/ML, Google Public Sector
207,000–301,000 / Year
Location
Reston, VA, USA
Experience
Senior
Posted
Jul 18, 2026
Apply by
August 17, 2026
Applicants
0
Early applicantEasy applyFull-timeWork from Office
Job Description
Minimum qualifications: Bachelor's degree or equivalent practical experience. 8 years of experience programming in C++, Java, Python, Kotlin or Go. Experience in technical leadership, including defining technical road maps, delivering projects, and maintaining code quality standards. Experience in parallel computing paradigms, hardware-level optimization, and low-level accelerator optimization. Preferred qualifications: Master's degree or PhD in a quantitative discipline (e.g., Computer Science, Physics, Applied Mathematics, or similar). 8 years of experience designing, building, and operating large-scale distributed data systems and production machine learning deployments. Experience deploying modern deep learning architectures using frameworks like PyTorch or TensorFlow on large-scale clusters. Experience with cloud-native infrastructure (Docker, Kubernetes) and managing distributed filesystems and cloud object storage. Active, or the ability to obtain, a Secret security clearance.
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. As a part of the technical Senior AI/ML Software Engineer, you will lead the architecture and deployment of large-scale distributed data systems and advanced machine learning pipelines. In this role, you will design infrastructure capable of analyzing high-throughput data streams. You will bridge the gap between High-Performance Computing (HPC) and modern AI applications, optimizing complex inference workloads for specialized hardware accelerators while guiding cross-functional engineering teams.Google Public Sector brings the magic of Google to the mission of government and education with solutions purpose-built for enterprises. We focus on helping United States public sector institutions accelerate their digital transformations, and we continue to make significant investments and grow our team to meet the complex needs of local, state and federal government and educational institutions.Individual pay is determined by factors including job-related skills, experience, and relevant education or training. US: $207000 - $301000 (USD) + 20% bonus target + equity + benefits Learn more about benefits at Google.
Responsibilities Architect and operate advanced data synthesis pipelines and AI-based retrieval applications. Manage petabyte-scale data ingestion and synchronization across compute environments, including local storage, cloud backends, and on-prem resources. Optimize highly parallel numerical operations and ML inference algorithms for specialized hardware accelerators. Lead technical direction and provide engineering mentorship for groups developing complex production software systems. Implement rigorous data life-cycle policies to ensure system resilience, data integrity, and fault recovery at scale.
Key Responsibilities
- Architect and operate advanced data synthesis pipelines and AI-based retrieval applications.
- Manage petabyte-scale data ingestion and synchronization across compute environments.
- Optimize highly parallel numerical operations and ML inference algorithms for specialized hardware accelerators.
- Lead technical direction and provide engineering mentorship for groups developing complex production software systems.
- Implement rigorous data life-cycle policies to ensure system resilience, data integrity, and fault recovery at scale.
Requirements
- Bachelor's degree or equivalent practical experience
Skills Required
C++JavaPythonKotlinGoParallel computing paradigmsHardware-level optimizationLow-level accelerator optimizationTechnical leadershipDistributed data systemsMachine learning deploymentsEngineering mentorshipCross-functional collaborationPyTorchTensorFlowDockerKubernetesCloud-native infrastructureDistributed filesystemsCloud object storage
Benefits
- 20% bonus target
- Equity
- Health insurance
- 401(k) match
- Unlimited PTO
- Parental leave
- Commuter benefits
- Tuition reimbursement
- Employee discounts
App exclusive · Free
Smart Job AI Coach
Your personal interview coach on every job — readiness tips, profile improvements, and role-specific prep. Available only in the Pulse Job app.
Interview readiness
See how prepared you are and what to improve for each role.
Personalized tips
Actionable suggestions based on your profile and the job.
After you apply
Keep coaching momentum from job detail through application success.
Similar roles for you
Matched using this role's title and skills. Open the job search anytime to see every listing.

Fullstack Staff Software Engineer - PAM
Okta
Full-timeHybrid
BengaluruSenior
Staff Software Engineer, Google Global Infrastructure Networking
Full-timeEasy applyWork from Office
BengaluruSenior
Staff Software Engineer, AI/ML and Moma Agentic
Full-timeEasy applyWork from Office
HyderabadSenior

Staff Software Engineer, QE
AlphaSense India
Full-timeEasy applyWork from Home
Remote - IndiaSenior

Staff Software Engineer, QE
AlphaSense
Full-timeEasy applyWork from Home
Remote - IndiaSenior
Staff Network Automation Engineer, Network Infrastructure and Capacity
176,000–257,000 / Year
Full-timeEasy applyWork from Office
AtlantaSenior