Implement neural hamiltonian flow as described in the [original paper](https://arxiv.org/pdf/1909.13789.pdf).