FreeGrad

github.com ↗

PyTorch library for custom backward passes, straight-through estimators and gradient transforms.