git+https://github.com/tinkoff-ai/d4rl@master#egg=d4rl
tqdm==4.64.0
wandb==0.12.21
mujoco-py==2.1.2.14
numpy==1.23.1
gym[mujoco_py]==0.23.0
torch==1.11.0
sortedcontainers==2.4.0
pyrallis==0.3.1
--find-links https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
jax[cuda11_cudnn82]
flax==0.6.1
optax==0.1.3
distrax==0.1.2
chex==0.1.5
h5py==3.6.0
