hessian_trace_estimation icon indicating copy to clipboard operation
hessian_trace_estimation copied to clipboard

Hessian trace estimation using PyTorch and Hutch++

Hessian trace estimation

A tiny repository containing just one notebook. The notebook shows how to efficiently approximate the trace of a Hessian, using automatic differentiation (PyTorch) and a recently published randomized algorithm called Hutch++.