Follow the instructions to install Jax with CUDA support if you want to use your GPU.
Then, simply install jwave
using pip
pip install jwave
You 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 jwave
Then, generate the environment using
make virtualenv
Install jax with GPU support (assumes CUDA > 11.1) using
make jaxgpu
Before using jwave
, activate the environment from the root folder of jwave using
source .venv/bin/activate