diff --git a/python/mlx/__init__.py b/python/mlx/__init__.py
new file mode 100644
index 000000000..8449aef19
--- /dev/null
+++ b/python/mlx/__init__.py
@@ -0,0 +1,4 @@
+# Include version
+__version__ = "0.0.6"
+import .optimizers
+import .nn