Merge branch 'develop' into add-comm-dtypes

axonn-ai · Feb 27, 2024 · 79a76aa · 79a76aa
2 parents d5a8ec5 + 6980466
commit 79a76aa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -18,5 +18,5 @@
     classifiers=["Development Status :: 2 - Pre-Alpha"],
     keywords="deep learning, distributed computing, parallel computing",
     packages=find_packages(),
-    install_requires=["torch", "mpi4py"],
+    install_requires=["torch"],
 )