Reinforcement Learning AI Engineer

Your growth matters to us - explore our career development opportunities.

Connect with others in our people-first culture and enhance our collective ingenuity.

Learn how we’ll support you as you pursue a balanced, fulfilling life.

Discover what to expect during your journey as a candidate with us.

Are you an innovative and experience d artifi cia l intelligence ( AI ) developer spe cia lizing 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 Petting Zoo, high-fidelity simulators, or custom environments.
• Apply RL techniques such as policy optimization, value-based learning, model-based RL, and imitation learning.
• Collaborate with domain experts to define reward structures, constraints, and evaluation met rics aligned with mission objectives.
• Implement distributed training workflows leveraging cloud compute, containerization, and orchestration technologies.
• Transition trained models into production systems, following strong sof tware engineering best practices.
• Contribute to system architecture and performance optimization in Python with opportunities to extend into C++ or Rus t for high-performance components.

Join us. The world can’t wait.

You Have:

• 3+ years of experience developing and training RL agents
• Experience with AI, data science, ML engineering, or sof tware engineering
• Experience with Gym or Petting Zoo interfaces
• Experience with ML frameworks such as PyTorch, Tensor Flow, or JAX
• Experience developing technical solutions using Rus t, Python, or C++
• Knowledge of RL and artifi cia l neural networks
• Ability to travel up to 25% of the time
• Bachelor's degree in a CS, AI, or Engineering field

Nice If You Have:

• Experience applying RL to autonomy, control systems, or mission-scale
• Experience with MARL
• Experience with AFSIM or other high-fidelity simulation environments
• Experience with embedded systems programming in Rus t, C, or C++
• Experience in GPU programming, including CUDA or RAPID
• Knowledge of modern sof tware design patterns, including microservice design and orchestration in Kubernetes deployment
• Master’s degree in CS, AI, Engineering, or a related field

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…