Reinforcement Learning AI Engineer
Booz Allen Hamilton6 hours ago
Huntsville, AL, USAMid Level
H1B sponsor
Base Salary
$99k - $225k/yr
Responsibilities
- Design, implement, and train reinforcement learning and multi-agent reinforcement learning algorithms for complex decision-making problems.
- Develop scalable training pipelines using Python and modern ML frameworks.
- Build and evaluate agents in Gym, PettingZoo, high-fidelity simulators, or custom environments.
- Apply policy optimization, value-based learning, model-based reinforcement learning, and imitation learning techniques.
- Define reward structures, constraints, and evaluation metrics with domain experts.
- Implement distributed training workflows using cloud compute, containerization, and orchestration technologies.
- Transition trained models into production systems and contribute to architecture and performance optimization.
- Extend high-performance components into C++ or Rust when appropriate.
Requirements
- At least 3 years of experience developing and training reinforcement learning agents.
- Experience in AI, data science, machine learning engineering, or software engineering.
- Experience with Gym or PettingZoo interfaces and ML frameworks such as PyTorch, TensorFlow, or JAX.
- Experience developing technical solutions with Rust, Python, or C++.
- Knowledge of reinforcement learning and artificial neural networks.
- Bachelor's degree in computer science, artificial intelligence, or engineering.
- Secret clearance is required.
- Preferred qualifications include RL experience in autonomy, control systems, or mission-scale applications; multi-agent reinforcement learning; AFSIM or other high-fidelity simulators; embedded programming; GPU programming with CUDA or RAPIDS; in-space solutions; Kubernetes deployment; and TS/SCI clearance.
- A master's degree in computer science, artificial intelligence, engineering, or a related field is preferred.
Benefits
- Hybrid work arrangement in Huntsville, Alabama, with additional locations in Colorado Springs, El Segundo, and McLean; hybrid employees are expected to work frequently from a Booz Allen facility and may visit customer facilities.
- Health, life, disability, financial, and retirement benefits.
- Paid leave, professional development, tuition assistance, work-life programs, and dependent care.
- Recognition awards and access to benefits for eligible full-time and part-time employees working at least 20 hours per week.
Tech Stack
Categories
About Booz Allen Hamilton
Booz Allen Hamilton builds and integrates software, analytics, AI, cloud, and cybersecurity solutions while providing engineering and consulting services, primarily for U.S. defense, intelligence, and civil government clients, with select commercial partnerships. Its business is largely government contracting and professional services delivering mission systems and cyber defense. Founded in 1914 and headquartered in McLean, Virginia, Booz Allen is a public company traded on the NYSE (ticker: BAH).