Nonlinear Model Predictive Control (NMPC) based on CVXPY and JAX in Python - View it on GitHub