control-jax A library for data-driven model predictive control using the JAX ecosystem. Existing libraries Below is a list of existing Python libraries for MPC: trajax mpcjax do-mpc hilo-mpc pyMPC nmpyc ddmpc python-control mpcpy Other interesting code: jax-control-algorithms mpc.pytorch gurobi-machinelearning