Add FP16 support for low-precision HVP gradient math, power iteration, and Lanczos

noahgolmant / pytorch-hessian-eigenthings

Efficient PyTorch Hessian eigendecomposition tools!

MIT License

360 stars 43 forks source link

Closed noahgolmant closed 3 years ago

noahgolmant commented 3 years ago

Adds flags for FP16 support in addition to other minor changes. Resolves #34.