The Opportunity:
Booz 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.
The successful candidate 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 grow within a large community of machine learning engineers across the company and collaborate with a team of dedicated Space industry and AI professionals to deliver world-class solutions to solve missile defense problems in near-real time.
What You'll Work On:
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.
Join us. The world can’t wait.
You Have:
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
Nice If You Have:
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
Clearance:
Applicants selected will be subject to a security investigation and may need to meet eligibility requirements for access to classified information.
Compensation
At Booz Allen, we celebrate your contributions, provide you with opportunities and choices, and support your total well-being. Our offerings include health, life, disability, financial, and retirement benefits, as well as paid leave, professional development, tuition assistance, work-life programs, and dependent care. Our recognition awards program acknowledges employees for exceptional performance and superior demonstration of our values. Full-time and part-time employees working at least 20 hours a week on a regular basis are eligible to participate in Booz Allen’s benefit programs. Individuals that do not meet the threshold are only eligible for select offerings, not inclusive of health benefits. We encourage you to learn more about our total benefits by visiting the Resource page on our Careers site and reviewing Our Employee Benefits page.
Salary at Booz Allen is determined by various factors, including but not limited to location, the individual’s particular combination of education, knowledge, skills, competencies, and experience, as well as contract-specific affordability and organizational requirements. The projected compensation range for this position is $55,200.00 to $126,000.00 (annualized USD). The estimate displayed represents the typical salary range for this position and is just one component of Booz Allen’s total compensation package for employees. This posting will close within 90 days from the Posting Date.Identity Statement
As part of the application process, you are expected to be on camera during interviews and assessments. We reserve the right to take your picture to verify your identity and prevent fraud.
Work Model
Our people-first culture prioritizes the benefits of flexibility and collaboration, whether that happens in person or remotely.
- If this position is listed as remote or hybrid, you’ll periodically work from a Booz Allen or client site facility.
- If this position is listed as onsite, you’ll work with colleagues and clients in person, as needed for the specific role.
Commitment to Non-Discrimination
All qualified applicants will receive consideration for employment without regard to disability, status as a protected veteran or any other status protected by applicable federal, state, local, or international law.
Top Skills
Similar Jobs
What you need to know about the San Francisco Tech Scene
Key Facts About San Francisco Tech
- Number of Tech Workers: 365,500; 13.9% of overall workforce (2024 CompTIA survey)
- Major Tech Employers: Google, Apple, Salesforce, Meta
- Key Industries: Artificial intelligence, cloud computing, fintech, consumer technology, software
- Funding Landscape: $50.5 billion in venture capital funding in 2024 (Pitchbook)
- Notable Investors: Sequoia Capital, Andreessen Horowitz, Bessemer Venture Partners, Greylock Partners, Khosla Ventures, Kleiner Perkins
- Research Centers and Universities: Stanford University; University of California, Berkeley; University of San Francisco; Santa Clara University; Ames Research Center; Center for AI Safety; California Institute for Regenerative Medicine
