Separate differentiation between descriptor and model
hghcomphys opened this issue · comments
Instead of relying entirely on the PyTorch gradient, this can take a heavy lift from the PyTorch and optimize the final performance.
hghcomphys / pantea
A Python library for developing machine learning interatomic potentials, based on Google JAX.