requirements.txt

jax>=0.2.13
jaxlib>=0.1.67
scipy>=1.4.1
numpy>=1.18.5
matplotlib~=3.4.2
pandas~=1.2.4
seaborn~=0.11.1
simple-parsing
cython
gin-config
absl-py
optax>=0.0.6
tqdm~=4.60.0
tensorboardX~=2.2
dm-haiku>=0.0.3
cyipopt

aiohttp==3.7.4.post0
async-timeout==3.0.1
attrs==21.2.0
bcrypt==3.2.0
certifi==2021.5.30
cffi==1.14.5
chardet==4.0.0
click==8.0.1
cloudpickle==1.6.0
configparser==5.0.2
cryptography==3.4.7
cycler==0.10.0
docker-pycreds==0.4.0
experiment_buddy
fabric==2.6.0
funcy==1.16
gitdb==4.0.7
GitPython==3.1.17
idna==2.10
importlib-metadata==4.5.0
invoke==1.5.0
kiwisolver==1.3.1
matplotlib==3.4.2
multidict==5.1.0
paramiko==2.7.2
pathlib2==2.3.5
pathtools==0.1.2
Pillow==8.2.0
promise==2.3
protobuf==3.17.2
psutil==5.8.0
pycparser==2.20
PyNaCl==1.4.0
pyparsing==2.4.7
python-dateutil==2.8.1
PyYAML==5.4.1
requests==2.25.1
scipy==1.7.0
sentry-sdk==1.1.0
shortuuid==1.0.1
six==1.16.0
smmap==4.0.0
subprocess32==3.5.4
tensorboardX==2.2
tqdm
typing-extensions==3.10.0.0
urllib3==1.26.5
wandb==0.10.31
yarl==1.6.3
zipp==3.4.1