diff --git a/requirements.txt b/requirements.txt
index 861d649..1beb5a9 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -13,7 +13,7 @@ pyarrow>=15.0.0
 pybind11
 pylatexenc
 ray[default]
-tensordict<0.6
+tensordict<0.9
 torchdata
 transformers
 # vllm==0.6.3.post1