torch-ema (conda): Tiny PyTorch library for maintaining a moving average of a collection of parameters.
Keywords: deep-learning|neural-networks|pytorch