This is your work, valued
RevTorch. Framework for creating (partially) reversible neural networks with PyTorch
PartiallyReversibleUnet. Python