PyWavelets
jax
jaxlib
scipy
matplotlib
numpy
pytest
nox

[jax_cpu]
jax
jaxlib
