Framework for creating (partially) reversible neural networks with PyTorch - View it on GitHub