Follow the instructions to install Jax with CUDA support if you want to use your GPU.
Then, simply install jwave using pip
pip install jwaveYou can use the provided make file to generate an existing environment that contains jwave.
First, clone the repository and move into its root directory
git clone git@github.com:ucl-bug/jwave.git
cd jwaveThen, generate the environment using
make virtualenvInstall jax with GPU support (assumes CUDA > 11.1) using
make jaxgpuBefore using jwave, activate the environment from the root folder of jwave using
source .venv/bin/activate