Installation
============

If you would like to use GPax with a GPU acceleration, follow these `instructions <https://github.com/google/jax#installation>`_ to install JAX with a GPU support.

Then, install GPax using pip as

.. code:: bash

    pip install gpax

for the stable version or

.. code:: bash
    
    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).