jinghong-zhang / ipie

ipie stands for Intelligent Python-based Imaginary-time Evolution with a focus on simplicity and speed.

ipie stands for Intelligent Python-based Imaginary-time Evolution with a focus on simplicity and speed.

ipie inherits a lot of QMC features from pauxy.

Copyright by Joonho Lee ([email protected]) Our first release paper can be found at https://pubs.acs.org/doi/10.1021/acs.jctc.2c00934

Features

ipie currently supports:

• estimation of the ground state energy of ab-initio systems using phaseless AFQMC with support for CPUs and GPUs.
• simple data analysis.
• other legacy functionalities available in pauxy such as the ground state and finite-temperature energies and properties (via backpropagation) of the ab initio, UEG, Hubbard, and Hubbard-Holstein models.

Installation

Linux and Mac OS wheels are available for installation via pip

$ pip install ipie

For develpment you can instead clone the repository

$ git clone https://github.com/linusjoonho/ipie.git

and run the following in the top-level ipie directory

$ pip install -r requirements.txt
$ pip install -e .

Requirements

To build ipie with MPI support (via mpi4py) do:

$ pip install -e .[mpi]

Note that mpi4py requires a working MPI installation to be built on your machine. This it is often the trickiest dependency to setup correctly.

One of the easiest ways (if you are using pip to install ipie wheels) is via conda:

conda install openmpi

which will just install the OpenMPI library. We refer users to the mpi4py documentation for alternative ways of building mpi4py and the required MPI library.

Further requirements are listed in requirements.txt.

GPU Support

Cupy is is required when running calculations on GPUs which can be install following the instructions here .

Cuda aware MPI may be installed via conda-forge.

Running the Test Suite

ipie contains unit tests and some longer driver tests that can be run using pytest by running:

$ pytest -v

in the base of the repo. Some longer parallel tests are also run through the CI. See .github/workflows/ci.yml for more details.

Documentation

Documentation and tutorials are available at readthedocs.