fix jit static arguments using jax pytree utils
hghcomphys opened this issue · comments
This was fixed for structure and acsf classes.
A Python library for developing machine learning interatomic potentials, based on Google JAX.
hghcomphys opened this issue · comments
This was fixed for structure and acsf classes.