AI Research Engineer - GPU Simulation

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 policies to the real aircraft.

What you'll do

• 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

What you'll need

• MSc or PhD in Computer Science, Machine Learning, Robotics, Control Engineering, or a closely related field
• Solid software engineering skills, writing clean and well-structured code in Python and/or languages like Rust or modern C++, and experience deploying software to production including testing and monitoring
• Depth in at least one of the following: GPU programming, high-performance simulation, physics or flight-dynamics modelling, or navigation and control
• Own high-quality, maintainable code and shape the shared library and ecosystem that the wider team builds on
• Care about our mission and want your work to have real operational impact

Nice to have

• Solid AI knowledge and can reason about how models are trained and evaluated
• Knowledge breadth across specialist areas, for example combining GPU programming with simulation environment design
• Deep GPU expertise, such as writing and optimising CUDA kernels, or expertise in libraries such as Taichi, NVIDIA Warp/CuteDSL, Triton or Helion

What they offer

• Competitive salary and VSOP options
• Relocation support: up to €2,500 and 4 weeks temporary accommodation
• Learning: €500/£450 yearly allowance
• Health & wellness: gym membership and mental health support
• Enhanced parental leave: 22 weeks fully paid for primary caregivers & 6 weeks for secondary caregivers
• Hands-on onboarding program with tooling and applications across the company