requirements.txt

absl-py==2.0.0
appdirs==1.4.4
blinker==1.7.0
brax
celluloid==0.2.0
certifi==2023.7.22
charset-normalizer==3.3.2
chex
click==8.1.7
cloudpickle==3.0.0
contextlib2==21.6.0
contourpy==1.2.0
cycler==0.12.1
distrax
dm-env==1.6
dm-haiku==0.0.9
dm-tree==0.1.8
docker-pycreds==0.4.0
etils==1.5.2
evosax==0.1.5
Flask==3.0.0
Flask-Cors==4.0.0
flax==0.7.1
fonttools==4.44.0
fsspec==2023.10.0
gitdb==4.0.11
GitPython==3.1.40
glfw==2.6.2
grpcio==1.59.2
gym==0.26.2
gym-notices==0.0.8
gymnax==0.0.6
idna==3.4
importlib-resources==6.1.1
itsdangerous==2.1.2
jax==0.4.7
-f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html jaxlib==0.4.7+cuda11.cudnn82
jaxopt==0.8.2
Jinja2==3.1.2
jmp==0.0.4
kiwisolver==1.4.5
markdown-it-py==3.0.0
MarkupSafe==2.1.3
matplotlib==3.8.1
mdurl==0.1.2
ml-collections==0.1.1
ml-dtypes==0.3.1
msgpack==1.0.7
mujoco==3.0.0
nest-asyncio==1.5.8
numpy==1.26.1
nvidia-cublas-cu11==11.11.3.6
nvidia-cuda-cupti-cu11==11.8.87
nvidia-cuda-nvcc-cu11==11.8.89
nvidia-cuda-nvrtc-cu11==11.8.89
nvidia-cuda-runtime-cu11==11.8.89
nvidia-cudnn-cu11==8.9.6.50
nvidia-cufft-cu11==10.9.0.58
nvidia-cusolver-cu11==11.4.1.48
nvidia-cusparse-cu11==11.7.5.86
nvidia-nccl-cu11==2.19.3
opt-einsum==3.3.0
optax==0.1.4
orbax-checkpoint
packaging==23.2
Pillow==10.1.0
plotly==5.18.0
protobuf==4.25.0
psutil==5.9.6
Pygments==2.16.1
PyOpenGL==3.1.7
pyparsing==3.1.1
python-dateutil==2.8.2
pytinyrenderer==0.0.14
PyYAML==6.0.1
requests==2.31.0
rich==13.6.0
scipy==1.11.3
sentry-sdk==1.34.0
setproctitle==1.3.3
six==1.16.0
smmap==5.0.1
tabulate==0.9.0
tenacity==8.2.3
tensorboardX==2.6.2.2
tensorstore==0.1.49
toolz==0.12.0
tqdm==4.66.1
trimesh==4.0.4
typing_extensions==4.8.0
urllib3==2.0.7
wandb==0.16.0
Werkzeug==3.0.1
zipp==3.17.0