momentumnet
Simple and efficient RevNet-Library for PyTorch with XLA and DeepSpeed support and parameter offload