diff --git a/burn-tch/Cargo.toml b/burn-tch/Cargo.toml
index b7c5f3a0f..1f96d3489 100644
--- a/burn-tch/Cargo.toml
+++ b/burn-tch/Cargo.toml
@@ -29,3 +29,6 @@ burn-autodiff = { path = "../burn-autodiff", version = "0.12.0", default-feature
 burn-tensor = { path = "../burn-tensor", version = "0.12.0", default-features = false, features = [
   "export_tests",
 ] }
+
+[package.metadata.docs.rs]
+features = [ "doc" ]
\ No newline at end of file