add extra_require with libs for running tests (#36)

.gitignore

@@ -8,6 +8,7 @@ __pycache__/
 
 # Metal libraries
 *.metallib
+venv/
 
 # Distribution / packaging
 python/mlx/share

docs/src/install.rst

@@ -60,7 +60,7 @@ For developing use an editable install:
 To make sure the install is working run the tests with:
 
 .. code-block:: shell
-
+  pip install ".[testing]"
   python -m unittest discover python/tests
 
 C++ API

setup.py

@@ -145,6 +145,9 @@ if __name__ == "__main__":
         package_dir=package_dir,
         package_data=package_data,
         include_package_data=True,
+        extras_require={
+            "testing": ["numpy", "torch"]
+        },
         ext_modules=[CMakeExtension("mlx.core")],
         cmdclass={"build_ext": CMakeBuild},
         zip_safe=False,