The optimized codes (both MATLAB and Python) are coming soon.
If you ask for the current version of the codes, please contact me at wanxinjin@gmail.com or jin279@purdue.edu
All technical details are in this paper: "Pontryagin Differentiable Programming: An End-to-End Learning and Control Framework", link: https://arxiv.org/pdf/1912.12970v1.pdf
Thanks and sincerely,
Wanxin