requirements.txt

absl-py==1.4.0
asttokens==2.2.1
backcall==0.2.0
cached-property==1.5.2
cffi==1.15.1
chex==0.1.6
comm==0.1.3
contourpy==1.0.7
cycler==0.11.0
debugpy==1.6.6
decorator==5.1.1
dm-tree==0.1.8
etils==1.1.0
executing==1.2.0
flax==0.6.8
fonttools==4.39.0
igraph==0.10.4
importlib-resources==5.12.0
ipykernel==6.22.0
ipython==8.11.0
jax==0.4.6
jaxlib==0.4.6+cuda11.cudnn82
jedi==0.18.2
jupyter_client==8.1.0
jupyter_core==5.3.0
kiwisolver==1.4.4
llvmlite==0.39.1
markdown-it-py==2.2.0
matplotlib==3.7.1
matplotlib-inline==0.1.6
mdurl==0.1.2
ml-dtypes==0.1.0
mpi4jax==0.3.14.post1
mpi4py==3.1.4
msgpack==1.0.5
nest-asyncio==1.5.6
NetKet==3.8
numba==0.56.4
numba4jax==0.0.10.post1
numpy==1.23.5
opt-einsum==3.3.0
optax==0.1.4
orbax==0.1.3
orbax-checkpoint==0.2.3
orjson==3.8.7
packaging==23.0
pandas==2.0.3
parso==0.8.3
pexpect==4.8.0
pickleshare==0.7.5
Pillow==9.4.0
platformdirs==3.1.1
plum-dispatch==1.7.4
prompt-toolkit==3.0.38
psutil==5.9.4
ptyprocess==0.7.0
pure-eval==0.2.2
pycparser==2.21
Pygments==2.14.0
pyparsing==3.0.9
python-dateutil==2.8.2
pytz==2023.3
PyYAML==6.0
pyzmq==25.0.2
rich==13.3.2
scipy==1.10.1
six==1.16.0
stack-data==0.6.2
tensorstore==0.1.36
texttable==1.6.7
toolz==0.12.0
tornado==6.2
tqdm==4.65.0
traitlets==5.9.0
typing_extensions==4.5.0
tzdata==2023.3
wcwidth==0.2.6