diff --git a/tensordict/nn/__init__.py b/tensordict/nn/__init__.py
index d1ce6e580..814f21533 100644
--- a/tensordict/nn/__init__.py
+++ b/tensordict/nn/__init__.py
@@ -11,3 +11,16 @@
     set_interaction_mode,
 )
 from .sequence import TensorDictSequential
+
+__all__ = [
+    "dispatch_kwargs",
+    "TensorDictModule",
+    "TensorDictModuleWrapper",
+    "get_functional",
+    "make_functional",
+    "repopulate_module",
+    "ProbabilisticTensorDictModule",
+    "ProbabilisticTensorDictSequential",
+    "set_interaction_mode",
+    "TensorDictSequential",
+]
diff --git a/tensordict/prototype/__init__.py b/tensordict/prototype/__init__.py
index cd752d6a0..3752c15c5 100644
--- a/tensordict/prototype/__init__.py
+++ b/tensordict/prototype/__init__.py
@@ -1 +1,6 @@
 from .tensorclass import is_tensorclass, tensorclass
+
+__all__ = [
+    "is_tensorclass",
+    "tensorclass",
+]