AL/ML Engineer, Lead

AL/ML Engineer, Lead

The Opportunity:

As an experienced engineer, you know that machine learning (ML) is critical to understanding and processing massive datasets. Your ability to conduct statistical analyses on business processes using ML techniques makes you an integral part of delivering a customer-focused solution. We need your technical knowledge and desire to problem-solve to support public health and safety. As an ML engineer on our team, you’ll train, test, deploy, and maintain models that learn from data.

In this role, you’ll own and define the direction of mission-critical solutions by applying best-fit ML algorithms and technologies. You’ll be part of a large community of ML engineers across the company and collaborate with data engineers, data scientists, solutions architects, enterprise architects, and product owners to deliver world-class solutions. You’ll help design and develop enterprise ML operations frameworks, support mentorship and training initiatives, process data and information at a massive scale, and perform A/B testing tasks on statistical models, ML algorithms, and systems.

Your advanced skills and technical expertise will guide clients as they navigate the evolving landscape of ML, Generative AI, AI agents, and retrieval-augmented generation (RAG) technologies. This position is located in Atlanta, GA.

What You'll Work On:

• Build and maintain complex data pipelines using PySpark and Palantir Foundry to support AI and analytical workloads.
• Design and implement ML workflows, including Generative AI, AI agents, MCP-enabled systems, and RAG pipelines for public health use cases.
• Develop robust AI evaluation strategies to ensure the quality, reliability, and performance of deployed AI solutions.
• Design AI architectures supporting agent memory, state management, retrieval optimization, and interaction with structured and unstructured public health data.
• Collaborate with cross-functional teams to translate mission requirements into scalable AI/ML solutions while maintaining technical documentation and governance standards.
• Define and enforce data quality, privacy, anonymization, auditing, and ethical AI standards in compliance with public health and government requirements.

Join us. The world can’t wait.

You Have:

• 5+ years of experience designing, developing, and deploying ML models and AI solutions using Python
• 5+ years of experience with Generative AI, LLMs, AI agents, or RAG applications in enterprise environments
• Experience developing AI evaluation strategies and working with MCP-enabled systems
• Experience with ML frameworks and libraries such as TensorFlow or PyTorch
• Experience with data engineering using PySpark, SQL, and Palantir Foundry, including Foundry AIP
• Experience with Prompt Buddy, Codex, Claude, or OpenEvidence in enterprise environments
• Experience deploying ML solutions at scale using MLOps tools such as MLflow, and cloud platforms such as Azure, Databricks, or Domino
• Ability to implement AI agent architectures, retrieval optimization, and memory or state management workflows
• Ability to obtain and maintain a Public Trust or Suitability/Fitness determination based on client requirements
• Bachelor's degree in CS, Engineering, or Data Science

Nice If You Have:

• Experience with open-source LLMs such as LLaMA, Gemma, or Phi
• Experience in deep learning, NLP, conversational AI, or computer vision
• Experience supporting healthcare, biomedical, or public health AI/ML initiatives and government public health data systems
• Experience with chatbot development or end-to-end AI application deployment
• Experience with containerization, orchestration, CI/CD, and full-stack ML application deployment
• Experience working in Agile environments using tools such as Jira
• Experience writing technical documentation and presenting technical solutions to broad audiences
• Ability to collaborate effectively in matrixed environments with cross-functional stakeholders
• Master’s degree

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.

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 $128,700.00 to $292,000.00 (annualized USD). The estimate displayed represents the typical salary range for this position and is just one component of Booz Allen’s total compensation package for employees. This posting will close within 90 days from the Posting Date.

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.