Revflow
Revflow is a framework for creating reversible neural networks. The following is written in tensorflow. For a PyTorch veriosn of the following, use the following - https://github.com/RobinBruegger/RevTorch
The following code is used for our reformers package.
Install
requires tensorflow >= 2.0.0