diff --git a/requirements.txt b/requirements.txt
index 78d0af566f7..30799894689 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -26,6 +26,7 @@ jinja2==3.0.3
 pytorch-lightning
 torchx
 torchrl==0.2.0
+tensordict==0.2.0
 ax-platform
 nbformat>=4.2.0
 datasets