For the ANI image we have used the Pytorch implementation of the ANI potentials - TorchANI - and use it as an ASE calculator. We then simply leverage the algorithms in ASE to drive task such as geometry optimizations and normal modes calculations.