NetKet is an open-source project delivering cutting-edge methods for the study of many-body quantum systems with artificial neural networks and machine learning techniques. It is a Python library built on JAX.
- Homepage: https://www.netket.org
- Citing: https://www.netket.org/cite/
- Documentation: https://netket.readthedocs.io/en/latest/index.html
- Tutorials: https://netket.readthedocs.io/en/latest/tutorials/gs-ising.html
- Examples: https://github.com/netket/netket/tree/master/Examples
- Source code: https://github.com/netket/netket
Installation and Usage
NetKet runs on MacOS and Linux. We recommend to install NetKet using
pip, but it can also be installed with
It is often necessary to first update
pip to a recent release (
>=20.3) in order for upper compatibility bounds to be considered and avoid a broken installation.
For instructions on how to install the latest stable/beta release of NetKet see the Get Started page of our website or run the following command (Apple M1 users, follow that link for more instructions):
pip install --upgrade pip pip install --upgrade netket
If you wish to install the current development version of NetKet, which is the master branch of this GitHub repository, together with the additional dependencies, you can run the following command:
pip install --upgrade pip pip install 'git+https://github.com/netket/netket.git#egg=netket[all]'
To speed-up NetKet-computations, even on a single machine, you
can install the MPI-related dependencies by using
[mpi] between square brackets.
pip install --upgrade pip pip install --upgrade "netket[mpi]"
We recommend to install NetKet with all it's extra dependencies, which are documented below.
However, if you do not have a working MPI compiler in your PATH this installation will most likely fail because
it will attempt to install
mpi4py, which enables MPI support in netket.
The latest release of NetKet is always available on PyPi and can be installed with
NetKet is also available on conda-forge, however the version available through
can be slightly out of date compared to PyPi.
To check what is the latest version released on both distributions you can inspect the badges at the top of this readme.
netket with pip, you can pass the following extra variants as square brakets. You can install several of them by separating them with a comma.
"[dev]": installs development-related dependencies such as black, pytest and testing dependencies
mpi4pyto enable multi-process parallelism. Requires a working MPI compiler in your path
tensorboardxto enable logging to tensorboard, and openfermion to convert the QubitOperators.
"[all]": Installs all extra dependencies
To enable MPI support you must install mpi4jax. Please note that we advise to install mpi4jax with the same tool (conda or pip) with which you install it's dependency
To check whever MPI support is enabled, check the flags
>>> import netket >>> netket.utils.mpi.available True
Installation on Windows
WARNING: Windows support is experimental, and you should expect suboptimal performance.
We suggest to use Windows Subsystem for Linux (WSL), on which you can install NetKet following the same instructions as above, and CUDA and MPI work as intended.
However, if you just want to quickly get started with NetKet, it is also possible to install it natively on Windows. First, download an unofficial
jaxlib wheel from cloudhan/jax-windows-builder:
pip install --upgrade pip pip install jaxlib -f https://whls.blob.core.windows.net/unstable/index.html
Alternatively, you may specify a wheel version with CUDA support.
Then install NetKet as usual:
pip install --upgrade netket
If you want MPI support, please follow the discussion in mpi4jax.
To get started with NetKet, we recommend you give a look at our tutorials page, by running them on your computer or on Google Colaboratory. There are also many example scripts that you can download, run and edit that showcase some use-cases of NetKet, although they are not commented.
If you want to get in touch with us, feel free to open an issue or a discussion here on GitHub, or to join the MLQuantum slack group where several people involved with NetKet hang out. To join the slack channel just accept this invitation