Booz Allen Hamilton · 5 hours ago
Reinforcement Learning AI Engineer
El Segundo, CA
Full-time
Hybrid
New Grad, Entry Level
$55K/yr - $126K/yrBooz Allen Hamilton 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.
ConsultingCyber SecurityIT InfrastructureManagement ConsultingSecurity
No H1B
Security Clearance RequiredU.S. Citizen Only
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 Gymnasium, 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
Qualification
Required
Experience in developing and training reinforcement learning agents
Experience in artificial intelligence, data science, machine learning engineering, or software engineering
Experience with developing technical solutions using Python, C++, or Rust
Knowledge of machine learning, including reinforcement learning
Ability to obtain a Secret clearance
Scheduled to obtain Bachelor's degree by Spring or Summer 2026
Preferred
Experience applying RL to autonomy, control systems, or mission-scale
Experience with Multi-Agent Reinforcement Learning (MARL)
Experience with Gym or Gymnasium environments
Experience with AFSIM or other high-fidelity simulation environments
Experience with embedded systems programming in C, C++, or Rust
Experience in GPU programming, including CUDA or RAPID
Experience with ML frameworks, such as PyTorch, TensorFlow, or JAX
Experience with developing in-space solutions
Knowledge of modern software design patterns, including microservice design and orchestration in Kubernetes deployment
Bachelor's degree in a Computer Science, Artificial Intelligence, or Engineering field
Benefits
Health
Life
Disability
Financial
Retirement benefits
Paid leave
Professional development
Tuition assistance
Work-life programs
Dependent care
Company
Booz Allen Hamilton
Booz Allen Hamilton is a consulting firm that specializes in analytics, technology, and engineering.
Founded in 1914
10001+ employees
Funding
Current Stage
Public CompanyTotal Funding
$3.03B2025-03-11Post Ipo Debt· $650M
2023-08-01Post Ipo Debt· $650M
2020-08-13Post Ipo Debt· $700M
Leadership Team
Matthew Calderone
Chief Financial Officer and Executive Vice President
Kristine Anderson
Chief Operating Officer
Recent News
WSJ.com: US Business
2026-01-24
Benzinga.com
2026-01-23
2026-01-23
Company data provided by crunchbase