/pytorch-kronecker-factorization

PyTorch Kronecker Factored Approximation of the Hessian

Primary LanguagePython

Stargazers