Reinforcement Learning AI EngineerThe Opportunity: Booz Allen is seeking an innovative and experienced AI developer specializing in reinforcement learning to join our growing team for Space solutions. In this role, you will leverage your expertise in artificial intelligence, data science, and machine learning engineering to train, test, deploy, and maintain models that learn from data. You will collaborate with cross-functional teams to translate reinforcement learning research into operational capability and production-grade code, bringing significant technological advancements that drive mission success. You’ll pioneer a growing community of machine learning engineers across the company. You’ll collaborate with a team of dedicated Space, Military, Intelligence, Engineering, and AI professionals to deliver bleeding-edge solutions to solve high-priority national defense problems.

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 policy optimization, 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.