fermiflow / FermiFlow

Ab-initio simulation of interacting fermions with equivariant normalizing flow.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

logo

FermiFlow

Build Status

The code requires python >= 3.6 and PyTorch >= 1.7.1. A GPU support is highly recommended. (Otherwise the code would likely be painfully slow!) The transformation of fermion coordinates is implemented as a continuous normalizing flow, where we have used the differentiable ODE solver torchdiffeq with O(1) memory consumption.

Run python BetaFermionHO2D.py --help to check out the available parameters and options for the finite-temperature variational Monte Carlo (VMC) code of a 2D quantum dot system. Below is a simple example:

python BetaFermionHO2D.py --beta 10.0 --nup 3 --Z 2.0 --deltaE 2.0 --cuda 0 --boltzmann --iternum 1000

The corresponding ground-state VMC code FermionHO2D.py is very similar.

To cite

@misc{xie2021abinitio,
      title={Ab-initio study of interacting fermions at finite temperature with neural canonical transformation}, 
      author={Hao Xie and Linfeng Zhang and Lei Wang},
      year={2021},
      eprint={2105.08644},
      archivePrefix={arXiv},
      primaryClass={cond-mat.str-el}
}

About

Ab-initio simulation of interacting fermions with equivariant normalizing flow.


Languages

Language:Jupyter Notebook 60.8%Language:Python 39.2%