requirements.txt

absl-py==2.0.0
appnope==0.1.3
asttokens==2.4.0
backcall==0.2.0
cached-property==1.5.2
chex==0.1.7
cloudpickle==3.0.0
clu==0.0.10
comm==0.1.4
contextlib2==21.6.0
contourpy==1.1.1
cycler==0.12.1
debugpy==1.8.0
decorator==5.1.1
diffrax==0.4.1
dm-tree==0.1.8
equinox==0.11.1
etils==1.5.1
executing==2.0.0
flax==0.7.4
fonttools==4.43.1
fsspec==2023.9.2
gast==0.5.4
importlib-resources==6.1.0
ipykernel==6.25.2
ipython==8.16.1
jax==0.4.14
jaxlib==0.4.14
jaxtyping==0.2.23
jedi==0.19.1
joblib==1.3.2
jupyter_client==8.4.0
jupyter_core==5.4.0
kiwisolver==1.4.5
markdown-it-py==3.0.0
matplotlib==3.8.0
matplotlib-inline==0.1.6
mdurl==0.1.2
ml-collections==0.1.1
ml-dtypes==0.3.1
msgpack==1.0.7
nest-asyncio==1.5.8
numpy==1.26.1
opt-einsum==3.3.0
optax==0.1.7
orbax-checkpoint==0.1.6
packaging==23.2
parso==0.8.3
pexpect==4.8.0
pickleshare==0.7.5
Pillow==10.1.0
platformdirs==3.11.0
prompt-toolkit==3.0.39
protobuf==4.24.4
psutil==5.9.6
ptyprocess==0.7.0
pure-eval==0.2.2
Pygments==2.16.1
pyparsing==3.1.1
python-dateutil==2.8.2
PyYAML==6.0.1
pyzmq==25.1.1
rich==13.6.0
scikit-learn==1.3.1
scipy==1.11.3
six==1.16.0
stack-data==0.6.3
tensorflow-probability==0.22.0
tensorstore==0.1.45
threadpoolctl==3.2.0
toolz==0.12.0
tornado==6.3.3
tqdm==4.66.1
traitlets==5.11.2
typeguard==2.13.3
typing_extensions==4.5.0
wcwidth==0.2.8
wrapt==1.15.0
zipp==3.17.0