Torchdiffeq Documentation, py for understanding how to use torchdiffeq to fit a simple spiral ODE.