├── .gitignore ├── .travis.yml ├── LICENSE ├── README.md ├── _assets └── density2D.gif ├── requirements.txt ├── src ├── BetaFermionHO2D.py ├── FermionHO2D.py ├── MLP.py ├── NeuralODE │ ├── __init__.py │ ├── function.py │ ├── nnModule.py │ └── utils.py ├── VMC.py ├── __init__.py ├── base_dist.py ├── equivariant_funs.py ├── flow.py ├── orbitals.py ├── potentials.py ├── slater.py └── utils.py └── tests ├── demos ├── checkpoint.py ├── misc.py └── nn_cuda.py ├── test_MLP.py ├── test_NeuralODE.py ├── test_basedist.py ├── test_cuda.py ├── test_distributions.py ├── test_equivariant_funs.py ├── test_odeint.py ├── test_slater.py └── test_utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/buwantaiji/FermiFlow/HEAD/.gitignore -------------------------------------------------------------------------------- /.travis.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/buwantaiji/FermiFlow/HEAD/.travis.yml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/buwantaiji/FermiFlow/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/buwantaiji/FermiFlow/HEAD/README.md -------------------------------------------------------------------------------- /_assets/density2D.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/buwantaiji/FermiFlow/HEAD/_assets/density2D.gif -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | numpy 2 | scipy 3 | -------------------------------------------------------------------------------- /src/BetaFermionHO2D.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/buwantaiji/FermiFlow/HEAD/src/BetaFermionHO2D.py -------------------------------------------------------------------------------- /src/FermionHO2D.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/buwantaiji/FermiFlow/HEAD/src/FermionHO2D.py -------------------------------------------------------------------------------- /src/MLP.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/buwantaiji/FermiFlow/HEAD/src/MLP.py -------------------------------------------------------------------------------- /src/NeuralODE/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/NeuralODE/function.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/buwantaiji/FermiFlow/HEAD/src/NeuralODE/function.py -------------------------------------------------------------------------------- /src/NeuralODE/nnModule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/buwantaiji/FermiFlow/HEAD/src/NeuralODE/nnModule.py -------------------------------------------------------------------------------- /src/NeuralODE/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/buwantaiji/FermiFlow/HEAD/src/NeuralODE/utils.py -------------------------------------------------------------------------------- /src/VMC.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/buwantaiji/FermiFlow/HEAD/src/VMC.py -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/base_dist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/buwantaiji/FermiFlow/HEAD/src/base_dist.py -------------------------------------------------------------------------------- /src/equivariant_funs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/buwantaiji/FermiFlow/HEAD/src/equivariant_funs.py -------------------------------------------------------------------------------- /src/flow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/buwantaiji/FermiFlow/HEAD/src/flow.py -------------------------------------------------------------------------------- /src/orbitals.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/buwantaiji/FermiFlow/HEAD/src/orbitals.py -------------------------------------------------------------------------------- /src/potentials.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/buwantaiji/FermiFlow/HEAD/src/potentials.py -------------------------------------------------------------------------------- /src/slater.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/buwantaiji/FermiFlow/HEAD/src/slater.py -------------------------------------------------------------------------------- /src/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/buwantaiji/FermiFlow/HEAD/src/utils.py -------------------------------------------------------------------------------- /tests/demos/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/buwantaiji/FermiFlow/HEAD/tests/demos/checkpoint.py -------------------------------------------------------------------------------- /tests/demos/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/buwantaiji/FermiFlow/HEAD/tests/demos/misc.py -------------------------------------------------------------------------------- /tests/demos/nn_cuda.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/buwantaiji/FermiFlow/HEAD/tests/demos/nn_cuda.py -------------------------------------------------------------------------------- /tests/test_MLP.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/buwantaiji/FermiFlow/HEAD/tests/test_MLP.py -------------------------------------------------------------------------------- /tests/test_NeuralODE.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/buwantaiji/FermiFlow/HEAD/tests/test_NeuralODE.py -------------------------------------------------------------------------------- /tests/test_basedist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/buwantaiji/FermiFlow/HEAD/tests/test_basedist.py -------------------------------------------------------------------------------- /tests/test_cuda.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/buwantaiji/FermiFlow/HEAD/tests/test_cuda.py -------------------------------------------------------------------------------- /tests/test_distributions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/buwantaiji/FermiFlow/HEAD/tests/test_distributions.py -------------------------------------------------------------------------------- /tests/test_equivariant_funs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/buwantaiji/FermiFlow/HEAD/tests/test_equivariant_funs.py -------------------------------------------------------------------------------- /tests/test_odeint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/buwantaiji/FermiFlow/HEAD/tests/test_odeint.py -------------------------------------------------------------------------------- /tests/test_slater.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/buwantaiji/FermiFlow/HEAD/tests/test_slater.py -------------------------------------------------------------------------------- /tests/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/buwantaiji/FermiFlow/HEAD/tests/test_utils.py --------------------------------------------------------------------------------