paramfittorchdemo Install pip install paramfittorchdemo You should checkout the writeup on fitting parameters for differential equations using pytorch here.