Export public module attributes

Author: tnadav

mlx_lm/__init__.py

@@ -9,3 +9,12 @@
 from .convert import convert
 from .generate import batch_generate, generate, stream_generate
 from .utils import load
+
+__all__ = [
+    "__version__",
+    "convert",
+    "batch_generate",
+    "generate",
+    "stream_generate",
+    "load",
+]