Reinforcement Learning AI Engineer
About The Position
The Opportunity : Booz Allen is seeking an innovative and experience d AI developer spe cia lizing in reinforcement learning to join our growing team for Space solutions. In this role, you will leverage your expertise in artifi cia l intelligence, data science, and machine learning engineering to train, test, deploy, and maintain models that learn from data. You will collaborate with cross-functional teams to translate reinforcement learning research into operational capability and production-grade code, bringing significant technological advancements that drive mission success. You’ll pioneer a growing community of machine learning engineers across the company. You’ll collaborate with a team of dedicated Space, Military, Intelligence, Engineering, and AI professionals to deliver bleeding-edge solutions to solve high-priority national defense problems.
Requirements
- Experience developing and training reinforcement learning agents
- Experience with Gym or PettingZoo interfaces
- Experience with ML frameworks such as PyTorch, TensorFlow, or JAX
- Experience with artifi cia l intelligence, data science, machine learning engineering, or sof t war e engineering
- Experience developing technical solutions using Python, C++, or Rus t
- Knowledge of reinforcement learning and artifi cia l neural networks
- Ability to obtain a Secret clearance
- Bachelor's degree in a Computer Science, Artifi cia l Intelligence, or Engineering field
Nice To Haves
- Experience applying RL to autonomy, control systems, or mission-scale
- Experience with Multi-Agent Reinforcement Learning ( MARL )
- Experience with AFSIM or other high-fidelity simulation environments
- Experience with embedded systems programming in C, C++, or Rus t
- Experience in GPU programming, including CUDA or RAPID
- Experience developing in-space solutions
- Knowledge of modern sof t war e design patterns, including microservice design and orchestration in Kubernetes deployment
- Master’s degree in Computer Science, Artifi cia l Intelligence, Engineering, or a related field
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 re war d structures, constraints, and evaluation met rics aligned with mission objectives.
- Implement distributed training workflows leveraging cloud compute, containerization, and orchestration technologies.
- Transition trained models into production systems, following strong sof t war e engineering best practices.
- Contribute to system architecture and performance optimization in Python with opportunities to extend into C++ or Rus t for high-performance components.
Benefits
- health, life, disability, financial, and retirement benefits, as well as paid leave, professional development, tuition assistance, work-life programs, and dependent care.
- recognition awards program acknowledges employees for exceptional performance and superior demonstration of our values.
Stand Out From the Crowd
Upload your resume and get instant feedback on how well it matches this job.
What This Job Offers
Job Type
Full-time
Career Level
Mid Level
Number of Employees
5,001-10,000 employees
