ACEsuit / mace

MACE - Fast and accurate machine learning interatomic potentials with higher order equivariant message passing.
Other
415 stars 157 forks source link

Calculate virials and stress with vectors to get atomic virial #325

Closed pobo95 closed 3 months ago

pobo95 commented 4 months ago

Description

Calculate virials and stress with vectors instead of displacement .

Motivation and Formula

Motivation To get atomic virial for NPT simulations and stress/atom in LAMMPS .

Formula image

image

How Has It Been Tested?

virials and stress ouput of vectors and displacement are consistent.

Warning

It occurs error when compiling model to torchscript jit.compile(lammps_model) seems to conflict with enisum. Do you have any hints about this problem?