higher

github.com ↗

higher is a pytorch library allowing users to obtain higher order gradients over losses spanning training loops rather than individual training steps.