RevTorch. Framework for creating (partially) reversible neural networks with PyTorch
PartiallyReversibleUnet. Python