AI Research Engineer - GPU Simulation

Job summary

Responsibilities

Join our GPU simulation team and build the simulator that trains our reinforcement-learning agents for air autonomy. You will shape the simulator that turns complex air-combat scenarios into learning environments at scale, model the physics behind air combat, and support the transfer of those…

• Build, extend, and scale our GPU-based simulator to train reinforcement-learning agents across thousands of parallel environments
• Develop the tooling that underpins RL training, including domain randomisation, to improve sim-to-real generalisation
• Improve the fidelity and performance of simulation models, from flight dynamics to sensing
• Enable sim-to-real transfer by supporting the export of trained policies and assets into downstream simulators and, ultimately, onto physical hardware
• Collaborate with AI specialists and engineers to align the simulator with training needs and mission-relevant outcomes

Tech stack

• Python
• Rust
• C++
• CUDA
• Triton