diff --git a/docs/Project.toml b/docs/Project.toml
index 8ec8cc050..e941459ad 100644
--- a/docs/Project.toml
+++ b/docs/Project.toml
@@ -45,7 +45,7 @@ LuxCUDA = "0.3.2"
 LuxCore = "1"
 LuxLib = "1"
 LuxTestUtils = "1.2"
-MLDataDevices = "1.1"
+MLDataDevices = "1.2"
 Optimisers = "0.3.3"
 Pkg = "1.10"
 Printf = "1.10"
diff --git a/docs/src/api/Accelerator_Support/MLDataDevices.md b/docs/src/api/Accelerator_Support/MLDataDevices.md
index 1009bed6c..d375c7fcb 100644
--- a/docs/src/api/Accelerator_Support/MLDataDevices.md
+++ b/docs/src/api/Accelerator_Support/MLDataDevices.md
@@ -18,6 +18,7 @@ MLDataDevices.gpu_backend!
 ```@docs
 MLDataDevices.cpu_device
 MLDataDevices.gpu_device
+MLDataDevices.xla_device
 ```
 
 ## Miscellaneous