There is a PyTorch implementation by the authors https://github.com/rtqichen/torchdiffeq https://github.com/rtqichen/torchdiffeq and there is a Julia implementa

There is a PyTorch implementation by the authors https://github.com/rtqichen/torchdiffeq https://github.com/rtqichen/torchdiffeq and there is a Julia implementation due: https://news.ycombinator.com/item?id=18676986#18691054 https://news.ycombinator.com/item?id=18676986#18691054. As for TensorFlow, we will have to wait and see if it can be done. The adjoint method is matrix multiplication, but the tricky part will also be integrating with numerical ODE solvers.