Small distributed launch helper (#1810)

2025-12-14 09:07:12 +08:00 · 2025-01-29 17:55:04 -08:00
parent 4758c8baa1
commit ded914f442
4 changed files with 338 additions and 26 deletions
--- a/setup.py
+++ b/setup.py
@@ -194,6 +194,7 @@ if __name__ == "__main__":
                "typing_extensions",
            ],
        },
+        entry_points={"console_scripts": ["mlx.launch = mlx.distributed_run:main"]},
        ext_modules=[CMakeExtension("mlx.core")],
        cmdclass={"build_ext": CMakeBuild, "generate_stubs": GenerateStubs},
        zip_safe=False,