absl-py>=2.0.0
chex>=0.1.5
cloudpickle>=3.0.0
distrax>=0.1.2
dm-tree>=0.1.7
flax>=0.9.0
hydra-core>=1.3.0
jax>=0.4.20
matplotlib>=3.8.0
numpy>=1.24.0
omegaconf>=2.2.0
optax>=0.1.5
orbax-checkpoint>=0.4.0
pandas>=2.0.0
property-cached>=1.6.4
PyYAML>=6.0
rliable>=1.0.7
scikit-learn>=1.0.0
SciencePlots==2.1.1
scipy>=1.10.0
seaborn>=0.12.0
statsmodels>=0.14.0
tensorflow-probability>=0.19.0
wandb>=0.15.0