Crazyflow simulator speeds up aerial robot training with realistic feedback

UWB Meets Crazyflow: Simulating Degraded Feedback at Scale for Aerial Robotics

Robotics

Summary

Simulating how drones behave is often slow and inaccurate. The authors created Crazyflow, a fast and precise simulation tool that runs many tests simultaneously on specialized computer hardware. They added a detailed model for radio and motion sensors that drones use, making the tests more realistic, especially when sensor signals are imperfect. This helps quickly train control programs for drones under real-world conditions.

What this means in practice

Authors

Martin Schuck, Marcel P. Rath, Yufei Hua, Abhishek Goudar, SiQi Zhou, Angela P. Schoellig

Abstract

In this work, we introduce Crazyflow, an accurate, differentiable simulator built on JAX. By leveraging jit compilation via XLA, Crazyflow unifies physics and control into a single differentiable computation graph, enabling massive parallelization on accelerated hardware without sacrificing modeling accuracy. This architecture achieves order-of-magnitude speedups over existing baselines, capable of training deployable reinforcement learning agents in seconds. To highlight its highly modular design, we demonstrate how easily Crazyflow can be extended by integrating a complete, high-fidelity Ultra-Wideband (UWB) and Inertial Measurement Unit (IMU) simulation pipeline coupled with a full-state Extended Kalman Filter (EKF). This capability allows for massive parallel controller evaluation under realistic, degraded state feedback with minimal impact on GPU throughput. By combining speed, accuracy, and extensibility, Crazyflow serves as a foundational tool for the next generation of aerial robotics research.