DMFF (Differentiable Molecular Force Field) is a Jax-based python package that provides a full differentiable implementation of molecular force field models.
GNU Lesser General Public License v3.0
154
stars
43
forks
source link
Add aux support in ml.py for MLP. Fix jit error when saving ml potential with save_dmff2tf.py #154