torchdiffeq
https://github.com/rtqichen/torchdiffeq
torchsde
https://github.com/google-research/torchsde
torchcde
https://github.com/patrick-kidger/torchcde
diffrax
https://github.com/patrick-kidger/diffrax
