PyWavelets
jax
scipy
matplotlib
click
numpy

[jax_cpu]
jax
jaxlib
