/hessian_trace_estimation

Hessian trace estimation using PyTorch and Hutch++

Primary LanguageJupyter NotebookApache License 2.0Apache-2.0

Watchers