diff --git a/ml/src/lib.rs b/ml/src/lib.rs
index bcf2f460d..26ff71d7f 100644
--- a/ml/src/lib.rs
+++ b/ml/src/lib.rs
@@ -1950,6 +1950,9 @@ pub mod prelude {
     // Constants
     pub use crate::{MAX_INFERENCE_LATENCY_US, PRECISION_FACTOR};
 
+    // GPU device management
+    pub use crate::gpu::{capabilities::GpuCapabilities, DeviceConfig};
+
     // Tensor types from candle
     pub use candle_core::{Device, Tensor};
     pub use candle_nn::{Module, VarBuilder, VarMap};