hghcomphys / pantea

A Python library for developing machine learning interatomic potentials, based on JAX.
https://pantea.readthedocs.io
Other
6 stars 2 forks source link

Implement Kalman filter #55

Closed hghcomphys closed 1 year ago

hghcomphys commented 1 year ago

Further investigation is necessary as it has been demonstrated that utilizing the Kalman filter as the updater rather than gradient descent leads to reduced network error. See this paper for more details.

hghcomphys commented 1 year ago

Done.