search

noahgolmant / pytorch-hessian-eigenthings

Efficient PyTorch Hessian eigendecomposition tools!
MIT License
360 stars 43 forks source link

Add fp16 support for half-precision power iteration and lanczos #34

Closed noahgolmant closed 3 years ago

noahgolmant commented 3 years ago

Add flags for fp16 support with