requirements.txt

gym == 0.23.1
mujoco_py == 2.1.2.14
numpy >= 1.20.2
dm_control >= 1.0.0
jax[cuda11_pip] == 0.3.15
-f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
jaxlib == 0.3.15
flax == 0.4.2
ml_collections >= 0.1.0
tqdm >= 4.60.0
optax == 0.1.3
absl-py == 1.3.0
scipy >= 1.6.0
wandb >= 0.12.14
glfw == 1.4.0
distrax==0.1.2
typing-extensions==4.3.0
chex==0.1.5
patchelf
moviepy
imageio
d4rl @ git+https://github.com/nakamotoo/D4RL.git