hessian_trace_estimation icon indicating copy to clipboard operation
hessian_trace_estimation copied to clipboard

Published 20 hours ago • verified publisher icon akshayka
→

Metadata

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++.

← Metadata

18
Stars
4
Forks
Watchers

Owner

verified publisher icon akshayka

Metadata

Hessian trace estimation using PyTorch and Hutch++

Back