/revflow

building blocks for Reversible Neural networks, written using Tensorflow

Primary LanguagePythonMIT LicenseMIT

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