Neural ODE ((Chen et. al. 2018)[https://arxiv.org/abs/1806.07366]), FFJORD ((Grathwohl et. al. 2019)[https://arxiv.org/abs/1810.01367], (repo)[https://github.com/rtqichen/ffjord]).
This repo contains wrappers and common architectures to learn dynamics.
Tests can be found under scripts/.