AI Research Engineer - GPU Simulation
Core
Build and scale a GPU-based simulator to train reinforcement-learning agents for air autonomy, modeling physics and enabling sim-to-real transfer for aircraft.
Role type
Senior IC AI Research Engineer (GPU Simulation)
Builds
High-fidelity GPU simulation environments for reinforcement learning training
Domain
Defense AI, GPU computing, physics-based simulation, flight dynamics
Deliverable
production ML models
Required skills
GPU programming, high-performance simulation, physics or flight-dynamics modelling, navigation and control, software engineering (Python/Rust/C++)
Preferred skills
CUDA kernel optimization, Taichi, NVIDIA Warp/CuteDSL, Triton, Helion, reinforcement learning theory
Technologies
Python, Rust, C++, CUDA, Taichi, NVIDIA Warp, CuteDSL, Triton, Helion
Responsibilities
Build and extend GPU-based simulators for thousands of parallel RL environments; Develop tooling for domain randomization and sim-to-real generalization; Improve fidelity and performance of flight dynamics and sensing models; Enable policy export to downstream simulators and physical hardware; Collaborate with AI specialists to align simulation with mission outcomes
Seniority
Senior, hands-on IC
