1 files changed, 27 insertions, 0 deletions
diff --git a/var/spack/repos/builtin/packages/py-backpack-for-pytorch/package.py b/var/spack/repos/builtin/packages/py-backpack-for-pytorch/package.py
new file mode 100644
index 0000000000..c8579d87ac
--- /dev/null
+++ b/var/spack/repos/builtin/packages/py-backpack-for-pytorch/package.py
@@ -0,0 +1,27 @@
+# Copyright 2013-2024 Lawrence Livermore National Security, LLC and other
+# Spack Project Developers. See the top-level COPYRIGHT file for details.
+#
+# SPDX-License-Identifier: (Apache-2.0 OR MIT)
+
+from spack.package import *
+
+
+class PyBackpackForPytorch(PythonPackage):
+    """BackPACK: Packing more into backprop."""
+
+    homepage = "https://github.com/f-dangel/backpack"
+    pypi = "backpack-for-pytorch/backpack-for-pytorch-1.6.0.tar.gz"
+
+    license("MIT")
+
+    version("1.6.0", sha256="af6495b71bacf82a1c7cab01aa85bebabccfe74d87d89f108ea72a4a0d384de3")
+
+    with default_args(type="build"):
+        depends_on("py-setuptools@38.3:")
+        depends_on("py-setuptools-scm")
+
+    with default_args(type=("build", "run")):
+        depends_on("py-torch@1.9:")
+        depends_on("py-torchvision@0.7:")
+        depends_on("py-einops@0.3:0")
+        depends_on("py-unfoldnd@0.2:0")