Remote AI/ML Tech Lead — Gen AI & RAG Systems

Booz Allen Hamilton

United States

Hybrid

USD 129,000 - 292,000

Full time

14 days+
Application generator

Don’t send a generic resume — generate a resume and cover letter tailored to this exact role.

Get past ATS filters

Benefits offered by this job

Health benefits
Tuition assistance
Work-life programs
Paid leave

Job summary

Booz Allen Hamilton seeks an AI/ML Technical Lead to define the technical direction of mission-critical AI systems supporting public health and safety. You will lead design and implementation of scalable ML and generative AI solutions, including AI agents, RAG pipelines, and evaluation frameworks that ensure reliability, safety, and mission alignment.

You will serve as a technical authority across a community of engineers, partnering with data engineers, architects, product owners, and

Qualifications

  • 5+ years designing, developing, and deploying AI/ML solutions using Python.

Responsibilities

  • Lead the design and development of scalable AI/ML systems for public health mission applications.
  • Architect generative AI solutions, including AI agents, MCP-enabled workflows, Prompt Buddy integrations, and RAG pipelines.
  • Define AI evaluation frameworks covering performance benchmarking, retrieval optimization, safety validation, and production monitoring.
  • Design and implement agent architectures with memory, state management, orchestration, and data interaction.
  • Integrate advanced AI tools such as Codex, Claude, and OpenEvidence into enterprise AI workflows.
  • Establish data governance, privacy, anonymization, documentation, and ethical AI standards across AI/ML systems.

Skills

Python
Generative AI
AI agents
RAG systems
Public health domain knowledge
Team leadership
PySpark
Palantir Foundry
MLOps
TensorFlow
PyTorch
Azure
Databricks

Education

Bachelor's degree

Tools

TensorFlow
PyTorch
MLflow
Azure
Databricks
Palantir Foundry
Codex
Claude
Prompt Buddy
OpenEvidence

Job description

Booz Allen Hamilton seeks an AI/ML Technical Lead to define the technical direction of mission-critical AI systems supporting public health and safety. You will lead design and implementation of scalable ML and generative AI solutions, including AI agents, RAG pipelines, and evaluation frameworks that ensure reliability, safety, and mission alignment.

You will serve as a technical authority across a community of engineers, partnering with data engineers, architects, product owners, and

Get your free, confidential resume review.

or drag and drop your file here.

Similar jobs

Similar jobs worth comparing

AI/ML Technical Lead: Generative AI & RAG for Public Health
AI/ML Technical Lead: Generative AI & RAG for Public Health

Booz Allen Hamilton • Atlanta (GA)

On-site
USD 128,000 - 292,000
Senior AI/ML Engineer: AI/ML Systems & NLP Leader
Senior AI/ML Engineer: AI/ML Systems & NLP Leader

Booz Allen Hamilton • Springfield (VA)

On-site
USD 134,000 - 181,000
AI/ML Engineering Lead — Enterprise Solutions
AI/ML Engineering Lead — Enterprise Solutions

Booz Allen Hamilton • Huntsville (AL)

Hybrid
USD 112,000 - 257,000
Remote AI/ML Engineer for Government AI
Remote AI/ML Engineer for Government AI

Socket.dev • McLean (VA)

On-site
USD 55,000 - 126,000
Remote AI Engineer — Scalable ML & LLMs
Remote AI Engineer — Scalable ML & LLMs

AIToolboard • Fort Belvoir (VA), Northern (KY)

Hybrid
USD 99,000 - 225,000
Health benefits
Life insurance
Disability insurance
+5
AI/ML Data Scientist & Team Leader (Remote)
AI/ML Data Scientist & Team Leader (Remote)

Booz Allen Hamilton • Kingstowne (VA)

On-site
USD 99,000 - 225,000
Senior AI/ML Engineer — NLP, LLMs & Production Systems
Senior AI/ML Engineer — NLP, LLMs & Production Systems

Booz Allen Hamilton • Alexandria (VA)

On-site
USD 99,000 - 225,000
Mission-Critical AI/ML Engineer
Mission-Critical AI/ML Engineer

Booz Allen Hamilton • Dayton (OH)

On-site
USD 99,000 - 225,000
Health benefits
Vision benefits
Life insurance
+2
Senior AI/ML Engineer - Mission-Critical, Remote
Senior AI/ML Engineer - Mission-Critical, Remote

Booz Allen Hamilton • Fort Meade (MD)

On-site
USD 99,000 - 225,000
Senior AI Applications Engineer — Remote
Senior AI Applications Engineer — Remote

Phase2 Technology • Honolulu (HI)

On-site
USD 86,000 - 198,000