Reinforcement Learning AI Engineer

Booz Allen Hamilton is seeking an innovative and experienced AI developer specializing in reinforcement learning. The role involves training, testing, deploying, and maintaining AI models, collaborating with teams to translate research into operational capabilities that drive significant technological advancements.

What you'll do:

• 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

What they're looking for:

• 3+ years of experience developing and training RL agents
• Experience with AI, data science, ML engineering, or software engineering
• Experience with Gym or PettingZoo interfaces
• Experience with ML frameworks such as PyTorch, TensorFlow, or JAX
• Experience developing technical solutions using Rust, Python, or C++
• Knowledge of RL and artificial neural networks
• Secret clearance
• Bachelor's degree in a CS, AI, or Engineering field
• 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 Rust, C, or C++

Benefits:

• Health, life, disability, financial, and retirement benefits
• Paid leave
• Professional development
• Tuition assistance
• Work-life programs
• Dependent care
• 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
• 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.