Hi,
This is not a library provided by the core developers so I don’t know.
Maybe the people that wrote that lib are reading this forum and will answer. Otherwise, maybe you should open an issue on the repo directly: https://github.com/rtqichen/torchdiffeq 72