aiqm / torchani

Accurate Neural Network Potential on PyTorch
https://aiqm.github.io/torchani/
MIT License
464 stars 128 forks source link

Use PyTorch autograd's hessian #532

Closed zasdfgbnm closed 3 years ago

zasdfgbnm commented 3 years ago

PyTorch now officially supports torch.autograd.functional.hessian, so there is no need to implement hessian in TorchANI, and vibrational_analysis should be adjusted to support the hessian returned by torch.autograd.functional.hessian.