AI​/ML Engineer, Public Sector

Overview

Google Public Sector in Reston, VA, USA;
Washington D.C., DC, USA. By applying you will have an opportunity to share your preferred working location:
Reston, VA, USA;
Washington D.C., DC, USA.

In this role, you will focus on ensuring the quality of customer engagements, while building excellent relationships with key stakeholders both internally and externally. These relationships will help your team meet the customer’s technical requirements and business objectives. You will deliver and execute leading by example to ensure customers in your region succeed.

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.

The US base salary range for this full-time position is $147,000-$216,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 benefits at Google.

• Act as a trusted technical advisor to customers and solve machine learning tests.
• Create and deliver recommendations, tutorials, blog articles, sample code, and technical presentations adapting to different levels of business and technical stakeholders.
• Work with Customers, Partners, and Google Product teams to deliver solutions into production.
• Coach customers on the practical tests in machine learning systems like feature extraction/feature definition, data validation, monitoring, and management of features/models.
• Travel up to 30% of the time in-region for meetings, technical reviews, and onsite delivery activities as needed.

• Bachelor’s degree in Engineering, Computer Science, related field, or equivalent practical experience.
• 6 years of experience with machine learning model development and deployment, frameworks (e.g., PyTorch, Tensor Flow, Jax, Ray), AI accelerators (e.g., TPUs, GPUs), model architectures (e.g., encoders, decoders, transformers), and using ML APIs.
• 6 years of experience in working with technical customers.
• Experience with Python.
• Active Top Secret/SCI US government security clearance.

• Experience in optimizing Large Language Model (LLMs) leveraging Retrieval Augmented Generation (RAG) architectures and fine-tuning techniques.
• Experience in building solutions for national security customers.
• Experience in working with multiple clouds.
• Experience in C++ programming language.
• Experience in Linux/Unix.
• Ability to lead design and implementation of AI-based solutions, web services, and debugging tools and containerize ML workloads.

Google is proud to be an equal opportunity and affirmative action employer. We are committed to building a workforce that is representative of the users we serve, creating a culture of belonging, and providing an equal employment opportunity regardless of race, creed, color, religion, gender, sexual orientation, gender identity/expression, national origin, disability, age, genetic information, veteran status, marital status, pregnancy or related condition (including breastfeeding), expecting or parents-to-be, criminal histories consistent with legal requirements, or any other basis protected by law.

See also Google's EEO Policy, Know your rights: workplace discrimination is illegal, Belonging at Google, and How we hire.

Google is a global company and, in order to facilitate efficient collaboration and communication globally, English proficiency is a requirement for all roles unless stated otherwise in the job posting.

To all recruitment agencies:
Google does not accept agency resumes. Please do not forward resumes to our jobs alias, Google employees, or any other organization location. Google is not responsible for any fees related to unsolicited resumes.