diff --git a/requirements/tpu.txt b/requirements/tpu.txt
index 7bb77c4a9..7ea239b48 100644
--- a/requirements/tpu.txt
+++ b/requirements/tpu.txt
@@ -11,6 +11,7 @@ ray[default]
 ray[data]
 setuptools==78.1.0
 nixl==0.3.0
+tpu_info==0.4.0
 
 # Install torch_xla
 --pre