update requirements

Author: Vincent Moens

.ci/docker/requirements.txt

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