WDM Wavelet transform

pywavelet is available on PyPI and can be installed with pip.

$ pip install pywavelet

Note: We have transforms availible in numpy, JAX and Cupy.

First set up a conda environment with python 3.10

$ mamba create -n pywavelet python=3.10

$ CONDA_OVERRIDE_CUDA=12.4  mamba install "jaxlib=*=*cuda*" jax -c conda-forge
$ CONDA_OVERRIDE_CUDA=12.4  conda install -c conda-forge cupy-core
$ pip install -e ".[dev]"
$ pre-commit install

Locate directory /tests from root directory. run

$ pytest .

Hopefully everything should run fine.