gradient-normalization
A practical implementation of GradNorm, Gradient Normalization for Adaptive Loss Balancing, in Pytorch