absl-py==1.4.0
jax==0.4.3
jaxlib==0.4.3
flax==0.6.4
numpy==1.24.2
orbax==0.1.1
tensorflow==2.11.0
tensorflow-probability==0.19.0
tensorboard==2.11.2
pandas==1.5.3
scikit-learn==1.3.1
scipy==1.10.1
matplotlib==3.7.2
