iit-DLSLab/Quadruped-PyMPC: A model predictive controller for quadruped robots based on the single rigid body model and written in python. Gradient-based (acados) or Sampling-based (jax) · GitHub

This repo contains a model predictive controller based on the single rigid body model and written in Python. It comes in two flavours: gradient-based via acados or sampling-based via jax. The controller is tested on real robots and is compatible with Mujoco.

Features gradient-based mpc:

• less than 5ms computation on an intel i7-13700H cpu
• optional integrators for model mismatch
• optional smoothing for the ground reaction forces
• optional foothold optimization
• optional real-time iteration or advanced-step real-time iteration
• optional zero moment point/center of mass constraints
• optional lyapunov-based criteria

Features sampling-based mpc:

• 10000 parallel rollouts in less than 2ms on an nvidia 4050 mobile gpu!
• optional step frequency adaptation for enhancing robustness
• implements different strategies: random sampling, mppi, or cemppi
• different control parametrizations: zero-order, linear splines or cubic splines (see mujoco-mpc)

Real-world deployment via:

• muse for state estimation
• unitree-ros2-dls for unitree robot communication

See here.

PRs are very welcome (search for TODO in the issue, or add what you like)!

If you find the work useful, please consider citing one of our works:

@INPROCEEDINGS{turrisi2024sampling,
  author={Turrisi, Giulio and Modugno, Valerio and Amatucci, Lorenzo and Kanoulas, Dimitrios and Semini, Claudio},
  booktitle={2024 IEEE/RSJ International Conference on Intelligent Robots and Systems (IROS)}, 
  title={On the Benefits of GPU Sample-Based Stochastic Predictive Controllers for Legged Locomotion}, 
  year={2024},
  pages={13757-13764},
  doi={10.1109/IROS58592.2024.10801698}}

@ARTICLE{elobaid2025adaptivestablempc,
  author={Elobaid, Mohamed and Turrisi, Giulio and Rapetti, Lorenzo and Romualdi, Giulio and Dafarra, Stefano and Kawakami, Tomohiro and Chaki, Tomohiro and Yoshiike, Takahide and Semini, Claudio and Pucci, Daniele},
  journal={IEEE Robotics and Automation Letters}, 
  title={Adaptive Non-Linear Centroidal MPC With Stability Guarantees for Robust Locomotion of Legged Robots}, 
  year={2025},
  volume={10},
  number={3},
  pages={2806-2813},
  doi={10.1109/LRA.2025.3536296}}

This repository is maintained by Giulio Turrisi and Daniel Ordonez.