f-dangel

BackPACK - 基于 PyTorch 构建的反向传播套件，能高效地计算梯度以外的量。