numpy
tensorflow
tensorflow-probability
