Search papers, labs, and topics across Lattice.
TurboMPC is a GPU-optimized differentiable model predictive control (MPC) solver that integrates sequential quadratic programming (SQP) and an alternating direction method of multipliers (ADMM) to efficiently handle complex constraints and costs in robotics applications. The solver achieves remarkable speed improvements, demonstrating up to 58 times faster performance than existing CPU and GPU differentiable solvers in various simulations, including constrained planning and reinforcement learning tasks. Additionally, TurboMPC's deployment in real-world scenarios, such as minimum-time racing, showcases its capability to significantly outperform hand-tuned MPC parameters through batched, GPU-accelerated Bayesian optimization.
Achieving up to 58 times faster performance than traditional solvers, TurboMPC revolutionizes the application of differentiable model predictive control in robotics.
Robotics increasingly relies on GPUs for parallel simulation, large-scale learning, and neural-network inference. For model predictive control (MPC) to scale with this paradigm, solvers must run efficiently on this hardware while remaining fast, differentiable, and compatible with expressive MPC formulations used in robotics. We present TurboMPC, a differentiable MPC solver that runs entirely on the GPU and supports state and control inequality constraints, implicit integrators, cross-time-coupled costs, and slack variables. TurboMPC combines sequential quadratic programming (SQP), an alternating direction method of multipliers (ADMM) inner solver, implicit differentiation, and a co-designed JAX-CUDA implementation for efficiency and ease of use. In simulation, we validate TurboMPC on constrained planning, humanoid imitation learning, and reinforcement learning with neural-network cost function tasks, achieving up to $15\times$ and $58\times$ speedups over state-of-the-art CPU and GPU differentiable solvers, respectively. We deploy TurboMPC on a full-scale car for minimum-time racing and find that batched, GPU-accelerated tuning of MPC parameters via Bayesian optimization yields significantly faster driving than a hand-tuned baseline. TurboMPC also scales to planning horizons of over $8000$ knot points while maintaining control of the vehicle. We open-source TurboMPC at: https://github.com/ToyotaResearchInstitute/turbompc
