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.
requires tensorflow >= 2.0.0