init

Author: Vincent Moens

.ci/docker/requirements.txt

@@ -28,9 +28,8 @@ tensorboard
 jinja2==3.1.3
 pytorch-lightning
 torchx
-# TODO: use stable 0.5 when released
--e git+https://github.com/pytorch/rl.git#egg=torchrl
--e git+https://github.com/pytorch/tensordict.git#egg=tensordict
+torchrl==0.5.0
+tensordict==0.5.0
 ax-platform
 nbformat>==5.9.2
 datasets