tensorflow
tensorflow-probability
PyWavelets
