diff --git a/setup.py b/setup.py
index 2a81f11cbab1..d3f9e85cd059 100644
--- a/setup.py
+++ b/setup.py
@@ -410,6 +410,7 @@ def run(self):
     + extras["modelcreation"]
     + extras["onnxruntime"]
     + extras["num2words"]
+    + deps_list("einops")
 )
 extras["dev-tensorflow"] = (
     extras["testing"]