First published version of the package.
It supplies utilities for computing gradient metrics on neural networks using PyTorch. Metrics are collected by utilizing backward hooks.
Currently the following metrics are supported:
PNorm
Min
Max
Mean
MeanStd
`GradientMetricCollector` is a helper class for collecting the metrics and computing the gradients.