Artificial Intelligence Engineer

Job Number: R0233275

Design and build robust knowledge frameworks, enabling AI systems to dynamically interact with structured and unstructured public health data, facilitating real‑time insights and decision‑making support. Leverage knowledge of the design and implementation of base architectures, data modeling for AI agents, memory and state management, and construction of RAG pipeline design to facilitate multi‑modal data integration partnering closely with data engineers on data pipeline coordination.

Support performance optimization of retrieval processes for public health use cases and define rigorous data quality and curation standards. Collaborate with AI teams and maintain thorough documentation of data architecture. Ensure compliance with data privacy standards, anonymization protocols, and ethical guidelines critical to public health data management. This position is located in Atlanta, GA.

• 4+ years of experience in AI/ML, generative AI, or knowledge‑based systems
• 2+ years of experience working with cloud platforms
• Experience in knowledge management and semantic search solutions
• Experience with vector databases such as Azure AI Search, Postgre
  
  SQL PgVector, Pinecone, Weaviate, or FAISS
• Experience in data engineering and scripting, including with Python, SQL, and API‑driven architectures
• Knowledge of enterprise search technologies, including Elasticsearch and Solr, and dense vector search methodologies
• Ability to obtain and maintain a Public Trust or Suitability/Fitness determination based on client requirements
• Bachelor’s degree

• Experience developing AI agents and multi‑agent architectures
• Experience with agent and orchestration frameworks such as Lang Chain, Llama Index, and Pydantic‑AI
• Experience in full‑stack development across front‑end and back‑end systems
• Experience designing knowledge graphs, ontologies, and semantic frameworks such as RDF, OWL, or Neo4j
• Experience in data governance, curation processes, and quality assurance practices within regulated health environments
• Experience in data‑centric AI/ML projects within healthcare, biomedical, or public health sectors
• Experience designing APIs and microservices architectures such as FastAPI or Flask, and caching mechanisms such as Redis
• Experience with cloud‑based AI data platforms such as Databricks, Azure AI Foundry, or GCP Vertex AI
• Knowledge of data ethics, privacy standards, security protocols, and emerging standards such as agent‑to‑agent protocols
• Possession of excellent communication skills, to facilitate collaboration between technical teams, public health experts, and stakeholders

Applicants selected will be subject to a government investigation and may need to meet eligibility requirements of the U.S. government client.

At Booz Allen, we celebrate your contributions, provide you with opportunities and choices, and support your total well‑being. Our offerings include health, life, disability, financial, and retirement benefits, as well as paid leave, professional development, tuition assistance, work‑life programs, and dependent care. Our recognition awards program acknowledges employees for exceptional performance and superior demonstration of our values. Full‑time and part‑time employees working at least 20 hours a week on a regular basis are eligible to participate in Booz Allen’s benefit programs.

Individuals that do not meet the threshold are only eligible for select offerings, not inclusive of health benefits. We encourage you to learn more about our total benefits by visiting the Resource page on our Careers site and reviewing Our Employee Benefits page. Salary at Booz Allen is determined by various factors, including but not limited to location, the individual’s particular combination of education, knowledge, skills, competencies, and experience, as well as contract‑specific affordability and organizational requirements.

The projected compensation range for this position is $99,000.00 to $ (annualized USD). The estimate displayed represents the typical salary range for this position and is just one component of Booz Allen’s total…