AL/ML Technical Lead

Booz Allen Hamilton•Atlanta, GA

1d•$128,700 - $292,000•Onsite

About The Position

As an AI/ML Technical Lead, you will define and drive the technical direction of mission-critical AI systems supporting Public Health and Safety. You will lead the design and implementation of scalable machine learning and generative AI solutions, including AI agents, retrieval-augmented generation (RAG) systems, 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 stakeholders to shape enterprise AI strategy, architecture standards, and delivery practices. Your work will span end-to-end AI system design—from data pipelines and modeling to agent orchestration, memory or state management, retrieval optimization, and production deployment. In this role, you will also lead the development of AI evaluation strategies, prompt engineering workflows, and governance frameworks using tools such as MCP, Prompt Buddy, Codex, Claude, and OpenEvidence. You will ensure systems are performant, explainable, secure, and compliant with public health data privacy and ethical standards.

Requirements

5+ years of experience designing, developing, and deploying machine learning and AI solutions using Python
5+ years of experience with Generative AI, LLMs, AI agents, or RAG systems in enterprise environments
3+ years of experience with AI agents, MCP frameworks, or AI evaluation strategy development
Experience with ML frameworks such as TensorFlow or PyTorch for production model development
Experience with data engineering using PySpark, SQL, and Palantir Foundry, including Foundry AIP
Experience with MLOps platforms such as MLflow and cloud environments such as Azure or Databricks
Experience integrating or working with AI tooling such as Codex, Claude, Prompt Buddy, or OpenEvidence in enterprise settings
Knowledge of public health, healthcare, or government data systems and associated governance practices
Ability to obtain and maintain a Public Trust or Suitability/Fitness determination based on client requirements
Bachelor's degree in Computer Science, Engineering, or Data Science

Nice To Haves

Experience working in healthcare, biomedical, or government public health AI/ML environments
Experience with open-source LLMs such as LLaMA, Gemma, or Phi
Experience with conversational AI, chatbot systems, or full-stack AI application development
Experience with containerization, CI/CD, orchestration, and production MLOps pipelines
Experience with Agile delivery environments and tools such as JIRA
Experience writing technical documentation and presenting AI/ML architecture decisions to a broad range of audiences
Knowledge of enterprise AI governance, compliance, and ethical AI frameworks
Knowledge of AI systems for retrieval, ranking, and decision-support in high-stakes environments
Ability to collaborate effectively in matrixed, cross-functional organization and design and deploy scalable AI systems involving retrieval optimization, memory or state management, and agent orchestration
Master's degree in Computer Science, Data Science, AI/ML, or a related field

Responsibilities

Lead the design and development of scalable AI/ML systems using PySpark and Palantir Foundry 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 structured or unstructured data interaction.
Integrate advanced AI tools such as Codex, Claude, and OpenEvidence into enterprise AI workflows and platforms.
Establish data governance, privacy, anonymization, documentation, and ethical AI standards across AI/ML systems.