Reinforcement Learning AI Engineer

What you'll do

• Design, implement, and train reinforcement learning and multi-agent reinforcement learning algorithms for complex decision-making.
• Develop scalable training pipelines using Python and modern machine learning frameworks.
• Build and evaluate agents in 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.
• 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, C++, or Rust.

What we're looking for

• 3+ years of experience developing and training RL agents.
• Experience with AI, data science, ML engineering, or software engineering.
• Experience with Gym or PettingZoo interfaces.
• Experience with ML frameworks such as PyTorch, TensorFlow, or JAX.
• Experience developing technical solutions using Rust, Python, or C++.
• Knowledge of RL and artificial neural networks.
• Bachelor's degree in a CS, AI, or Engineering field.
• Secret clearance.