# JAX libraries with GPU support.
jax==0.2.28
jaxlib @ https://storage.googleapis.com/jax-releases/cuda11/jaxlib-0.1.76+cuda11.cudnn82-cp38-none-manylinux2010_x86_64.whl
dm-haiku==0.0.5
chex==0.0.8
dm-env==1.2
dm-tree==0.1.5
optax==0.1.3
distrax==0.1.1

# other libs
scikit-learn==1.0.1
pydantic==1.8.2
python-dotenv==0.19.2
d4rl @ git+https://github.com/rafapi/d4rl.git@master#egg=d4rl
gym==0.18.3
mujoco-py==2.1.2.14
matplotlib==3.4.2
tqdm==4.61.2
wandb==0.12.1
