NN-based mean-field method - View it on GitHub