Remote RL AI Engineer — MARL & Production Systems

Aimlroles; United States; On-site; USD 99,000 - 225,000; Full time

Benefits offered by this job

Health benefits

Professional development

Job summary

Booz Allen Hamilton is seeking an innovative AI/ML engineer specialized in reinforcement learning to design, train, test, and deploy RL and MARL models that drive mission-critical impact. You will collaborate with space, military, intelligence, and AI professionals to translate research into production solutions.

You will build scalable training pipelines, evaluate agents in simulated environments, and transition models into production while adhering to software engineering best practices.

Qualifications

• 1+ 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 STEM field.

Responsibilities

• Design, implement, and train RL and MARL algorithms for complex decision-making.
• Develop scalable training pipelines using Python and modern ML frameworks.
• Build and evaluate agents in simulated environments (Gym/PettingZoo or high-fidelity simulators).
• Apply RL techniques including policy optimization, value-based learning, model-based RL, and imitation learning.
• Collaborate with domain experts to define reward structures and evaluation metrics.
• Implement distributed training workflows leveraging cloud, containers, and orchestration.
• Transition trained models into production systems following software engineering best practices.
• Contribute to system architecture and performance optimization in Python with potential CUDA/C++ extensions.

Skills

Reinforcement learning

AI/ML engineering

Python programming

Gym interfaces

PettingZoo interfaces

ML frameworks (PyTorch, TensorFlow, JX

Rust / C++ development

Secret clearance

Education

Bachelor's degree in STEM

Tools

Gym

PettingZoo

PyTorch

TensorFlow

JAX

Rust

C++

Python

Job description

Booz Allen Hamilton is seeking an innovative AI/ML engineer specialized in reinforcement learning to design, train, test, and deploy RL and MARL models that drive mission-critical impact. You will collaborate with space, military, intelligence, and AI professionals to translate research into production solutions.

You will build scalable training pipelines, evaluate agents in simulated environments, and transition models into production while adhering to software engineering best practices.