add modular imports to the init

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",
+]