Remote Databricks Architect Jobs in Georgia (NOW HIRING)

Remote Work:
Hybrid
Job Number:
R0243412
Location:
Atlanta,GA,US
Share job via:
Share
AL/ML Technical Lead
The Opportunity:
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 ML 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. This position is located in Atlanta, GA.
What You'll Work On:

• 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.

Join us. The world can't wait.
You Have:

• 5+ years of experience designing, developing, and deploying AI/ML 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, and managing a technical team
• 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

Nice If You Have:

• 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 organizations and design and deploy scalable AI systems involving retrieval optimization, memory or state management, and agent orchestration
• Master's degree in CS, Data Science, AI/ML, or a related field

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.
Not ready to apply? Join our Talent Community and sign up for job alerts.