RL AI Engineer - Production ML & MARL

Booz Allen Hamilton; Huntsville (AL); On-site; USD 99,000 - 225,000; Full time

Job summary

Booz Allen Hamilton is seeking an innovative Reinforcement Learning AI Engineer to design, train, test, deploy, and maintain RL agents that drive mission-critical impact across national-security domains.

You will collaborate with space, military, intelligence, engineering, and AI professionals to translate RL research into production-grade code, building scalable training pipelines in Python and extending to Rust or C++ for high-performance components.

Qualifications

• 3+ years of RL 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.
• Ability to travel up to 25% of the time.
• Bachelor's degree in CS, AI, or Engineering.

Responsibilities

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

Skills

Reinforcement learning

Multi-agent RL

Python programming

AI data science

Software engineering

Education

Bachelor's degree in CS/AI/Engineering

Tools

Gym

PettingZoo

PyTorch

TensorFlow

JAX

Rust

C++

Job description

Booz Allen Hamilton is seeking an innovative Reinforcement Learning AI Engineer to design, train, test, deploy, and maintain RL agents that drive mission-critical impact across national-security domains.

You will collaborate with space, military, intelligence, engineering, and AI professionals to translate RL research into production-grade code, building scalable training pipelines in Python and extending to Rust or C++ for high-performance components.