deepmodeling / DMFF

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
143 stars 41 forks source link

Add aux support in ml.py for MLP. Fix jit error when saving ml potential with save_dmff2tf.py #154

Closed dingye18 closed 5 months ago