Installation

If you would like to use GPax with a GPU acceleration, follow these instructions to install JAX with a GPU support.

Then, install GPax using pip as

pip install gpax

for the stable version or

pip install git+https://github.com/ziatdinovmax/gpax

to get the most recent updates.

If you are a Windows user, we recommend to use the Windows Subsystem for Linux (WSL2).