jax>=0.2.24
jaxlib>=0.1.73
tensorflow
tqdm>=4.48.2
tensorflow-probability[jax]
numpy
scipy
matplotlib
