diff --git a/src/models/__init__.py b/src/models/__init__.py
new file mode 100644
index 0000000000000000000000000000000000000000..557a2455780363cb23c91ff78e43ceeaa3cf9422
--- /dev/null
+++ b/src/models/__init__.py
@@ -0,0 +1,9 @@
+# This __init__.py exposes our SmallCNN and WideResNet classes
+# to code that imports from the models package.
+
+from .small_cnn import SmallCNN
+from .wide_resnet import WideResNet
+
+# By listing them in __all__, other files can do:
+# from models import SmallCNN, WideResNet
+__all__ = ["SmallCNN", "WideResNet"]