hghcomphys / pantea

A Python library for developing machine learning interatomic potentials, based on Google JAX.

https://pantea.readthedocs.io

fix jit static arguments using jax pytree utils

hghcomphys opened this issue 2 years ago · comments

Hossein Ghorbanfekr commented 2 years ago

Hossein Ghorbanfekr commented 2 years ago

This was fixed for structure and acsf classes.