numpy
tensorflow
tensorflow_probability
