diff --git a/examples/xr-transformer-neurips21/requirements.txt b/examples/xr-transformer-neurips21/requirements.txt
index 1184969..c2aa183 100644
--- a/examples/xr-transformer-neurips21/requirements.txt
+++ b/examples/xr-transformer-neurips21/requirements.txt
@@ -20,7 +20,7 @@ sentencepiece==0.1.96
 six==1.16.0
 threadpoolctl==2.2.0
 tokenizers==0.10.3
-torch==1.9.1
+torch==2.8.0
 tqdm==4.66.3
 transformers==4.11.0
 typing-extensions==3.10.0.2