diff --git a/pdm.lock b/pdm.lock
index 2857e7a..b78a746 100644
--- a/pdm.lock
+++ b/pdm.lock
@@ -5,7 +5,7 @@
 groups = ["default", "data", "dev", "lint", "nb", "tests"]
 strategy = ["inherit_metadata"]
 lock_version = "4.5.0"
-content_hash = "sha256:4d0f09206ae74c600eb4c435609c6ecb39fd98bde594aff1e3e0f72d33297997"
+content_hash = "sha256:24af6601a7e3ff5c6715e3fbf3db2b6fbf7c8707e6633011fc2ecfcade9dbe0a"
 
 [[metadata.targets]]
 requires_python = ">=3.11,<3.14"
@@ -851,6 +851,17 @@ files = [
     {file = "dopt_basics-0.2.4.tar.gz", hash = "sha256:c21fbe183bec5eab4cfd1404e10baca670035801596960822d0019e6e885983f"},
 ]
 
+[[package]]
+name = "einops"
+version = "0.8.2"
+requires_python = ">=3.9"
+summary = "A new flavour of deep learning operations"
+groups = ["dev"]
+files = [
+    {file = "einops-0.8.2-py3-none-any.whl", hash = "sha256:54058201ac7087911181bfec4af6091bb59380360f069276601256a76af08193"},
+    {file = "einops-0.8.2.tar.gz", hash = "sha256:609da665570e5e265e27283aab09e7f279ade90c4f01bcfca111f3d3e13f2827"},
+]
+
 [[package]]
 name = "execnet"
 version = "2.1.2"
diff --git a/pyproject.toml b/pyproject.toml
index ad95be5..d8ed34f 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -160,6 +160,7 @@ dev = [
     "huggingface-hub>=1.6.0",
     "tqdm>=4.66.5",
     "peft>=0.18.1",
+    "einops>=0.8.2",
 ]
 nb = [
     "jupyterlab>=4.3.5",