This repository provides an implementation in torch of the mean flow algorithm for solving the Fokker-Planck equation in high dimension.
Please install the packages listed in requirements.txt
The main implentation of Mean Flows for Fokker Planck Equation is contained in the 'training.py' file. Additionally, we have included two experiments (anharmonic and harmonic.py) that compare this learned result with the standard Euler-Maruyama approach.