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