Use **params instead of explicit attributes inside the trainer
hghcomphys opened this issue · comments
This make the design more generic.
Done.
A Python library for developing machine learning interatomic potentials, based on Google JAX.
hghcomphys opened this issue · comments
This make the design more generic.
Done.