Reinforcement Learning AI Engineer

The Opportunity:

Are you an innovative and experienced artificial intelligence (AI)developer specializing in reinforcement learning? Utilize your expertise in AI, data science, and machine learning (ML) engineering to train, test, deploy, and maintain models that learn from data to drive real-world mission critical impact. You will collaborate with dynamic teams to translate reinforcement learning research into operational capability and production-grade code, bringing significant technological advancements that drive mission success.

You’ll join a growing community of ML engineers focused on delivering products and solutions to our customers’ most challenging problems. You’ll collaborate with a team of dedicated space, military, intelligence, engineering, and AI professionals to deliver cutting-edge solutions to solve issues of national importance.

What You'll Work On:

• Design, implement, and train reinforcement learning (RL)and multi-agent reinforcement learning (MARL)algorithms for complex decision-making problems.

• Develop scalable training pipelines using Python and modern ML frameworks.

• Build and evaluate agents in simulated environments using Gym or PettingZoo, high-fidelity simulators, or custom environments.

• Apply RL techniques such as policyoptimization, value-based learning, model-based RL, and imitation learning.

• Collaborate with domain experts to define reward structures, constraints, and evaluation metrics aligned with mission objectives.

• Implement distributed training workflows leveraging cloud compute, containerization, and orchestration technologies.

• Transition trained models into production systems, following strong software engineering best practices.

• Contribute to system architecture and performance optimization in Python with opportunities to extend into C++ or Rust for high-performance components.

Join us. The world can’t wait.

You Have:

• 3+ years of experiencedeveloping and training RL agents

• Experiencewith AI, data science, ML engineering, or software engineering

• Experiencewith Gym or PettingZoo interfaces

• Experiencewith ML frameworks such as PyTorch, TensorFlow, or JAX

• Experiencedeveloping technical solutions using Rust, Python, or C++

• Knowledge ofRL and artificial neural networks

• Secret clearance

• Bachelor's degreein a CS, AI, or Engineering field

Nice If You Have:

• Experienceapplying RL to autonomy, control systems, or mission-scale

• Experiencewith MARL

• Experiencewith AFSIM or otherhigh-fidelity simulation environments

• Experiencewith embedded systems programming in Rust, C, or C++

• Experiencein GPU programming, including CUDA or RAPID

• Experiencedeveloping in-space solutions

• Knowledge ofmodern software design patterns, including microservice design and orchestration in Kubernetes deployment

• TS/SCI clearance

• Master’s degreein CS, AI, Engineering, or a relatedfield

Clearance:

Applicants selected will be subject to a security investigation and may need to meet eligibility requirements for access to classified information;Secret clearance is required.

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.