Artificial Intelligence Software Engineer Jobs in Georgia

The Opportunity:

Design and build robust knowledge frameworks, enabling AI systems to dynamically interact with structured and unstructured public health data, and facilitating real-time insights and decision-making support. Leverage knowledge of the design and implementation of base architectures, data modeling for AI agents, and Memory and State Management, and the 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.

You Have:

• 1+ years of experience in AI/ML, Generative AI, or knowledge-based systems
• 1+ years of experience with cloud platforms
• Experience in knowledge management and semantic search
• Experience with vector databases, including Azure AI Search, PostgreSQL PgVector, Pinecone, Weaviate, or FAISS
• Experience in data engineering and scripting, including with Python, SQL, and API-driven architectures
• Knowledge of enterprise search technologies such as 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

Nice If You Have:

• Experience developing AI Agents and multi-agent architectures
• Experience with agent and orchestration frameworks, including LangChain, LlamaIndex, and Pydantic-AI
• Experience with full-stack development across front-end and back-end systems
• Experience designing APIs and microservices architectures, including FastAPI or Flask, and caching mechanisms, including Redis
• Experience in data-centric AI/ML projects within healthcare, biomedical, or public health sectors
• Experience working with modern data platforms, including Databricks, Apache Spark, and Snowflake
• Experience in data governance, curation processes, and quality assurance practices within regulated health environments
• Knowledge of cloud-based AI data platforms such as Azure Foundry, AWS Bedrock, or GCP Vertex AI
• Possession of excellent communication skills, for facilitating collaboration between technical teams, public health experts, and stakeholders

Vetting:

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

Compensation

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.

Identity Statement

As part of the hiring process, we will ask you to complete an identity verification process that leverages advanced biometrics and artificial intelligence to ensure authenticity and protect against identity fraud. You are expected to be on camera during interviews and assessments. We reserve the right to take your picture to verify your identity and prevent fraud.

Candidate AI Usage Policy

AI is a part of our daily work at Booz Allen, and we are committed to the responsible and ethical use of AI tools. However, we want to ensure a fair candidate process based on your own skills and knowledge. As part of this commitment, the use of artificial intelligence (AI) or other tools to assist with responses during interviews (whether in-person or virtual) is prohibited unless permission is explicitly provided.

Work Model
Our people-first culture prioritizes the benefits of collaboration, whether it occurs in person or virtually. To support engagement and effective communication, employees working virtually are generally expected to have their cameras on during meetings.

• Remote: If this position is listed as remote, there may still be occasions when you are required to work in person at a Booz Allen or customer facility.

• Hybrid: If this position is listed as hybrid, you will be expected to work from a Booz Allen facility frequently, in alignment with leadership expectations and the needs of the role. You may also be required to work from or visit a customer facility.

• Onsite: If this position is listed as onsite, work will primarily be performed at a Booz Allen office or customer facility, where employees will collaborate directly with colleagues and customers as required by the role.

Commitment to Non-Discrimination

All qualified applicants will receive consideration for employment without regard to disability, status as a protected veteran or any other status protected by applicable federal, state, local, or international law.