Reinforcement Learning AI Engineer

Booz Allen Hamilton

Confirmed live yesterday High trust

Quick summary

Work type
On-site
Location
Huntsville, ALMcLean, VAEl Segundo, CAColorado Springs, CO
Salary
$99,000–$225,000 / yr
Posted
50 days ago
Freshness
Confirmed live yesterday

Market check

Salary context

Below market

How this pay compares to similar roles

Similar $197k
This role $162k
$82k most similar roles pay here $259k

This role pays less than 70% of similar roles. Most pay $162,000–$231,900 — the shaded band above. At the midpoint, this role pays about $162k versus about $197k for comparable roles.

Based on 240 similar postings.

Employer

About Booz Allen Hamilton

Booz Allen Hamilton is a management and technology consulting firm that provides analytics, digital, engineering, and cybersecurity solutions primarily to U.S. government agencies and commercial clients. Industry: Management & Technology Consulting

Booz Allen Hamilton currently has 802 open roles on FindRole.

Listed pay typically runs $86,800–$198,000 across 783 roles with salary data.

Most-posted roles

View all roles at Booz Allen Hamilton

At a glance

TL;DR · Reinforcement Learning AI Engineer

As a Reinforcement Learning AI Engineer, you will join a team of machine learning engineers to develop and deploy models that solve complex decision-making problems for mission-critical applications. You will design, implement, and train reinforcement learning and multi-agent reinforcement learning algorithms while building scalable training pipelines. Your daily work involves developing agents in simulated environments using Gym or PettingZoo, defining reward structures with domain experts, and transitioning trained models into production systems. The role requires proficiency in Python, PyTorch, TensorFlow, or JAX, along with experience in C++ or Rust for high-performance components. You will utilize cloud compute, containerization, and orchestration technologies to manage distributed workflows. This position focuses on solving technical challenges within the space, military, and intelligence sectors, specifically addressing problems related to autonomy and control systems.

What you'll do

  • Design and train reinforcement learning and multi-agent reinforcement learning algorithms for complex decision-making problems.
  • Develop scalable training pipelines using Python and modern machine learning frameworks like PyTorch, TensorFlow, or JAX.
  • Build and evaluate agents within simulated environments using Gym, PettingZoo, or custom high-fidelity simulators.
  • Apply specific RL techniques including policy optimization, value-based learning, model-based RL, and imitation learning.
  • Define reward structures, constraints, and evaluation metrics in collaboration with domain experts.
  • Implement distributed training workflows utilizing cloud compute, containerization, and orchestration technologies.
  • Transition trained models into production systems following rigorous software engineering best practices.
  • Optimize system architecture and performance using Python, C++, or Rust for high-performance components.

What we're looking for

  • Bachelor's degree in CS, AI, Engineering, or a related field.
  • Secret clearance is required.
  • 3+ years of experience developing and training RL agents.
  • Experience with ML frameworks such as PyTorch, TensorFlow, or JAX.
  • Experience with Gym or PettingZoo interfaces.
  • Proficiency in programming languages including Python, Rust, or C++.
  • Knowledge of reinforcement learning, artificial neural networks, and multi-agent systems.
  • Experience in AI, data science, machine learning engineering, or software engineering.

More like this

Similar roles

AI Engineer

Booz Allen Hamilton

Arlington, VA 71 days ago $77,600$176,000
LLMs RAG LangChain LlamaIndex OpenAI Anthropic Docker AWS Azure GCP RESTful APIs CI/CD Git Ollama Prompt Engineering Embeddings Containerization

Senior Deep Reinforcement Learning Engineer, Autonomous Driving

Nvidia

Santa Clara, CA 15 days ago $224,000$356,500
Reinforcement Learning PyTorch TensorFlow Python C++ Deep Reinforcement Learning PPO GRPO Actor-Critic Imitation Learning Generative Models Flow Matching Diffusion Autonomous Driving Simulation Environments Data Flywheels High-Performance Computing
10+ yrs exp

Machine Learning Engineer, Agentic AI

Apple Inc

Sunnyvale, CA 124 days ago $150,400$277,600
LLMs Agentic Systems Machine Learning Computer Vision Data Pipelines System Design Monitoring Observability