Add einops installation to setup.py

setup.py

@@ -168,5 +168,9 @@ setup(
     ],
     ext_modules=ext_modules,
     cmdclass={"build_ext": BuildExtension} if ext_modules else {},
-    python_requires=">=3.7"
+    python_requires=">=3.7",
+    install_requires=[
+        "torch",
+        "einops",
+    ],
 )