admf

See the repo here.

Implementation of traditional mean field approach in condensed matter physics but with AD infrastructure (powered by jax engine).

ADMF enables simple code and swift implementation for complicated quantum models.