gradient-centralization
optimizer & lr scheduler & loss function collections in PyTorch
Implement Gradient Centralization in TensorFlow