diff --git a/pkgs/development/python-modules/einops/default.nix b/pkgs/development/python-modules/einops/default.nix
index a24f75f1a8ee..f159fbd4f00f 100644
--- a/pkgs/development/python-modules/einops/default.nix
+++ b/pkgs/development/python-modules/einops/default.nix
@@ -2,7 +2,6 @@
   lib,
   buildPythonPackage,
   fetchFromGitHub,
-  fetchpatch2,
   hatchling,
   jupyter,
   nbconvert,
@@ -11,11 +10,12 @@
   pillow,
   pytestCheckHook,
   pythonOlder,
+  torch,
 }:
 
 buildPythonPackage rec {
   pname = "einops";
-  version = "0.8.0";
+  version = "0.8.1";
   pyproject = true;
 
   disabled = pythonOlder "3.7";
@@ -24,18 +24,9 @@ buildPythonPackage rec {
     owner = "arogozhnikov";
     repo = pname;
     tag = "v${version}";
-    hash = "sha256-6x9AttvSvgYrHaS5ESKOwyEnXxD2BitYTGtqqSKur+0=";
+    hash = "sha256-J9m5LMOleHf2UziUbOtwf+DFpu/wBDcAyHUor4kqrR8=";
   };
 
-  patches = [
-    # https://github.com/arogozhnikov/einops/pull/325
-    (fetchpatch2 {
-      name = "numpy_2-compatibility.patch";
-      url = "https://github.com/arogozhnikov/einops/commit/11680b457ce2216d9827330d0b794565946847d7.patch";
-      hash = "sha256-OKWp319ClYarNrek7TdRHt+NKTOEfBdJaV0U/6vLeMc=";
-    })
-  ];
-
   nativeBuildInputs = [ hatchling ];
 
   nativeCheckInputs = [
@@ -45,6 +36,7 @@ buildPythonPackage rec {
     parameterized
     pillow
     pytestCheckHook
+    torch
   ];
 
   env.EINOPS_TEST_BACKENDS = "numpy";
@@ -61,13 +53,16 @@ buildPythonPackage rec {
     "test_all_notebooks"
     "test_dl_notebook_with_all_backends"
     "test_backends_installed"
+    # depends on tensorflow, which is not available on Python 3.13
+    "test_notebook_2_with_all_backends"
   ];
 
-  disabledTestPaths = [ "tests/test_layers.py" ];
+  disabledTestPaths = [ "einops/tests/test_layers.py" ];
 
   __darwinAllowLocalNetworking = true;
 
   meta = with lib; {
+    changelog = "https://github.com/arogozhnikov/einops/releases/tag/${src.tag}";
     description = "Flexible and powerful tensor operations for readable and reliable code";
     homepage = "https://github.com/arogozhnikov/einops";
     license = licenses.mit;