requirements.txt

absl-py==1.4.0
asttokens==2.2.1
backcall==0.2.0
cached-property==1.5.2
chex==0.1.6
contourpy==1.0.7
cycler==0.11.0
decorator==5.1.1
dm-tree==0.1.8
etils==1.1.0
executing==1.2.0
flax==0.6.7
fonttools==4.39.0
importlib-resources==5.12.0
ipdb==0.13.13
ipython==8.11.0
jax==0.4.6
jaxlib==0.4.6
jedi==0.18.2
joblib==1.2.0
kiwisolver==1.4.4
markdown-it-py==2.2.0
matplotlib==3.7.1
matplotlib-inline==0.1.6
mdurl==0.1.2
msgpack==1.0.5
numpy==1.24.2
opt-einsum==3.3.0
optax==0.1.4
orbax==0.1.3
packaging==23.0
parso==0.8.3
pexpect==4.8.0
pickleshare==0.7.5
Pillow==9.4.0
prompt-toolkit==3.0.38
ptyprocess==0.7.0
pure-eval==0.2.2
Pygments==2.14.0
pyparsing==3.0.9
python-dateutil==2.8.2
PyYAML==6.0
rich==13.3.2
scikit-learn==1.2.2
scipy==1.10.1
six==1.16.0
stack-data==0.6.2
tensorstore==0.1.33
threadpoolctl==3.1.0
toolz==0.12.0
traitlets==5.9.0
typing_extensions==4.5.0
wcwidth==0.2.6