diff --git a/nanochat/__init__.py b/nanochat/__init__.py
index e69de29..3dcad8b 100644
--- a/nanochat/__init__.py
+++ b/nanochat/__init__.py
@@ -0,0 +1,33 @@
+# Import all submodules used by scripts
+from . import common
+from . import tokenizer
+from . import checkpoint_manager
+from . import core_eval
+from . import gpt
+from . import dataloader
+from . import loss_eval
+from . import engine
+from . import dataset
+from . import report
+from . import adamw
+from . import muon
+from . import configurator
+from . import execution
+
+# Make submodules available
+__all__ = [
+    "common",
+    "tokenizer",
+    "checkpoint_manager",
+    "core_eval",
+    "gpt",
+    "dataloader",
+    "loss_eval",
+    "engine",
+    "dataset",
+    "report",
+    "adamw",
+    "muon",
+    "configurator",
+    "execution",
+]