py-lightning-uq-box: add new package (#47132)

15 files changed, 334 insertions, 43 deletions

diff --git a/var/spack/repos/builtin/packages/py-asdfghjkl/package.py b/var/spack/repos/builtin/packages/py-asdfghjkl/package.py
new file mode 100644
index 0000000000..c9aab6fa6c
--- /dev/null
+++ b/var/spack/repos/builtin/packages/py-asdfghjkl/package.py
@@ -0,0 +1,24 @@
+# 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 PyAsdfghjkl(PythonPackage):
+    """ASDL: Automatic Second-order Differentiation (for Fisher, Gradient covariance, Hessian,
+    Jacobian, and Kernel) Library."""
+
+    homepage = "https://github.com/kazukiosawa/asdl"
+    pypi = "asdfghjkl/asdfghjkl-0.1a4.tar.gz"
+
+    license("MIT")
+
+    version("0.1a4", sha256="a934411a0ffdee6fcdccb19672196498ea6a8e55e3e67abbe67200c84b46ddee")
+
+    depends_on("py-setuptools@42:")
+
+    with default_args(type=("build", "run")):
+        depends_on("py-torch@1.13:")
+        depends_on("py-numpy")
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")
diff --git a/var/spack/repos/builtin/packages/py-curvlinops-for-pytorch/package.py b/var/spack/repos/builtin/packages/py-curvlinops-for-pytorch/package.py
new file mode 100644
index 0000000000..8197e8291e
--- /dev/null
+++ b/var/spack/repos/builtin/packages/py-curvlinops-for-pytorch/package.py
@@ -0,0 +1,29 @@
+# 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 PyCurvlinopsForPytorch(PythonPackage):
+    """scipy Linear operators for curvature matrices in PyTorch."""
+
+    homepage = "https://github.com/f-dangel/curvlinops"
+    pypi = "curvlinops_for_pytorch/curvlinops_for_pytorch-2.0.0.tar.gz"
+
+    license("MIT")
+
+    version("2.0.0", sha256="01f9925db9454fc9b0a31c7b83fc8ec2534c2eb12b7de7825a5298fc14e460e7")
+
+    with default_args(type="build"):
+        depends_on("py-setuptools@61:")
+        depends_on("py-setuptools-scm")
+
+    with default_args(type=("build", "run")):
+        depends_on("py-backpack-for-pytorch@1.6:1")
+        depends_on("py-torch@2:")
+        depends_on("py-scipy@1.7.1:1")
+        depends_on("py-tqdm@4.61:4")
+        depends_on("py-einops")
+        depends_on("py-einconv")
diff --git a/var/spack/repos/builtin/packages/py-einconv/package.py b/var/spack/repos/builtin/packages/py-einconv/package.py
new file mode 100644
index 0000000000..6a6c23dacd
--- /dev/null
+++ b/var/spack/repos/builtin/packages/py-einconv/package.py
@@ -0,0 +1,25 @@
+# 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 PyEinconv(PythonPackage):
+    """Convolutions as tensor contractions (einsums) for PyTorch."""
+
+    homepage = "https://github.com/f-dangel/einconv"
+    pypi = "einconv/einconv-0.1.0.tar.gz"
+
+    license("MIT")
+
+    version("0.1.0", sha256="6b103881b1268e43d581f285da4fa72b073c95f31b92575133bafed9929b6d98")
+
+    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")
+        depends_on("py-einops")
diff --git a/var/spack/repos/builtin/packages/py-ema-pytorch/package.py b/var/spack/repos/builtin/packages/py-ema-pytorch/package.py
index 00b8d48bde..dadf3aaecf 100644
--- a/var/spack/repos/builtin/packages/py-ema-pytorch/package.py
+++ b/var/spack/repos/builtin/packages/py-ema-pytorch/package.py
@@ -16,7 +16,9 @@ class PyEmaPytorch(PythonPackage):
 
     license("MIT", checked_by="alex391")
 
+    version("0.7.3", sha256="de640f1d1a054c79607aebfcfd4b8dfff1fba1110bf0c8f7d37517637450938a")
     version("0.5.1", sha256="e825212a44e8faae5d2cf2a1349961c4416cba0496ffa64d37718d8b06f206b2")
 
     depends_on("py-setuptools", type="build")
+    depends_on("py-torch@2:", when="@0.7:", type=("build", "run"))
     depends_on("py-torch@1.6:", type=("build", "run"))
diff --git a/var/spack/repos/builtin/packages/py-gpytorch/package.py b/var/spack/repos/builtin/packages/py-gpytorch/package.py
index c4a98153f6..a8bdb28538 100644
--- a/var/spack/repos/builtin/packages/py-gpytorch/package.py
+++ b/var/spack/repos/builtin/packages/py-gpytorch/package.py
@@ -19,6 +19,7 @@ class PyGpytorch(PythonPackage):
 
     license("MIT")
 
+    version("1.13", sha256="f4a488633a2a7a4ab37d12553d1d1dd39690043dbceef14ca428b7d5f89f73ba")
     version("1.10", sha256="6dc978ab9fbf220a845a4f1ea13104180fc50e6934081f421b37f6120afb7f18")
     version("1.9.1", sha256="0bdbba6f6d5957a0f43ef6dc7fec39c47e8a55f632ca33760c6189f259b3ccc3")
     version("1.9.0", sha256="a0608184c18a1f518d6a102473427abf00f5351421e12a934530953f6887b34b")
@@ -30,21 +31,28 @@ class PyGpytorch(PythonPackage):
     version("1.2.0", sha256="fcb216e0c1f128a41c91065766508e91e487d6ffadf212a51677d8014aefca84")
     version("1.1.1", sha256="76bd455db2f17af5425f73acfaa6d61b8adb1f07ad4881c0fa22673f84fb571a")
 
-    depends_on("python@3.8:", when="@1.9:", type=("build", "run"))
-    depends_on("python@3.7:", when="@1.7:", type=("build", "run"))
-    depends_on("python@3.6:", type=("build", "run"))
-    depends_on("py-setuptools", type="build")
-    depends_on("py-setuptools-scm", when="@1.9:", type="build")
-    depends_on("py-torch@1.11:", when="@1.9:", type=("build", "run"))
-    depends_on("py-torch@1.10:", when="@1.7:", type=("build", "run"))
-    depends_on("py-torch@1.9:", when="@1.6:", type=("build", "run"))
-    depends_on("py-torch@1.8.1:", when="@1.5:", type=("build", "run"))
-    depends_on("py-torch@1.7:", when="@1.3:", type=("build", "run"))
-    depends_on("py-torch@1.6:", when="@1.2:", type=("build", "run"))
-    depends_on("py-torch@1.5:", type=("build", "run"))
-    depends_on("py-scikit-learn", when="@1.2:", type=("build", "run"))
-    depends_on("py-linear-operator@0.1.1:", when="@1.9:", type=("build", "run"))
-    depends_on("py-linear-operator@0.2.0:", when="@1.9.1:", type=("build", "run"))
-    depends_on("py-linear-operator@0.4.0:", when="@1.10:", type=("build", "run"))
-    depends_on("py-numpy", when="@1.7:1.8", type=("build", "run"))
-    depends_on("py-scipy", when="@1.2:1.8", type=("build", "run"))
+    with default_args(type="build"):
+        depends_on("py-setuptools")
+        depends_on("py-setuptools-scm", when="@1.9:")
+
+    with default_args(type=("build", "run")):
+        depends_on("py-torch@2:", when="@1.13:")
+        depends_on("py-torch@1.11:", when="@1.9:")
+        depends_on("py-torch@1.10:", when="@1.7:")
+        depends_on("py-torch@1.9:", when="@1.6:")
+        depends_on("py-torch@1.8.1:", when="@1.5:")
+        depends_on("py-torch@1.7:", when="@1.3:")
+        depends_on("py-torch@1.6:", when="@1.2:")
+        depends_on("py-torch@1.5:")
+        depends_on("py-jaxtyping@0.2.19", when="@1.13:")
+        depends_on("py-mpmath@0.19:1.3", when="@1.12:")
+        depends_on("py-scikit-learn", when="@1.2:")
+        depends_on("py-scipy@1.6:", when="@1.13:")
+        depends_on("py-scipy", when="@1.2:1.8")
+        depends_on("py-linear-operator@0.5.3:", when="@1.13:")
+        depends_on("py-linear-operator@0.1.1:", when="@1.9:")
+        depends_on("py-linear-operator@0.2.0:", when="@1.9.1:")
+        depends_on("py-linear-operator@0.4.0:", when="@1.10:")
+
+        # Historical dependencies
+        depends_on("py-numpy", when="@1.7:1.8")
diff --git a/var/spack/repos/builtin/packages/py-h5py/package.py b/var/spack/repos/builtin/packages/py-h5py/package.py
index 4322037507..16d5506f8d 100644
--- a/var/spack/repos/builtin/packages/py-h5py/package.py
+++ b/var/spack/repos/builtin/packages/py-h5py/package.py
@@ -18,6 +18,8 @@ class PyH5py(PythonPackage):
     license("BSD-3-Clause")
 
     version("master", branch="master")
+    version("3.12.1", sha256="326d70b53d31baa61f00b8aa5f95c2fcb9621a3ee8365d770c551a13dbbcbfdf")
+    version("3.12.0", sha256="00955a079e9f86c5ae2cd08accb54396c69cda87152312ddd1528e3f90acc866")
     version("3.11.0", sha256="7b7e8f78072a2edec87c9836f25f34203fd492a4475709a18b417a33cfb21fa9")
     version("3.10.0", sha256="d93adc48ceeb33347eb24a634fb787efc7ae4644e6ea4ba733d099605045c049")
     version("3.9.0", sha256="e604db6521c1e367c6bd7fad239c847f53cc46646f2d2651372d05ae5e95f817")
@@ -40,11 +42,12 @@ class PyH5py(PythonPackage):
     version("2.5.0", sha256="9833df8a679e108b561670b245bcf9f3a827b10ccb3a5fa1341523852cfac2f6")
     version("2.4.0", sha256="faaeadf4b8ca14c054b7568842e0d12690de7d5d68af4ecce5d7b8fc104d8e60")
 
-    depends_on("c", type="build")  # generated
+    depends_on("c", type="build")
 
     variant("mpi", default=True, description="Build with MPI support")
 
     # Python versions
+    depends_on("python@3.9:", type=("build", "run"), when="@3.12:")
     depends_on("python@:3.9", type=("build", "run"), when="@:2.8")
 
     # Build dependencies
@@ -60,9 +63,9 @@ class PyH5py(PythonPackage):
     depends_on("py-setuptools", type="build")
 
     # Build and runtime dependencies
-    depends_on("py-numpy@1.17.3:", type=("build", "run"), when="@3.9:")
-    depends_on("py-numpy@1.19.3:", type=("build", "run"), when="@3:3.5 ^python@3.9.0:")
+    depends_on("py-numpy@1.19.3:", type=("build", "run"), when="@3:3.5,3.12: ^python@3.9.0:")
     depends_on("py-numpy@1.17.5:", type=("build", "run"), when="@3:3.5 ^python@3.8.0:3.8")
+    depends_on("py-numpy@1.17.3:", type=("build", "run"), when="@3.9:3.11")
     depends_on("py-numpy@1.14.5:", type=("build", "run"), when="@3:")
     depends_on("py-numpy@1.7:", type=("build", "run"), when="@:2")
     # https://github.com/h5py/h5py/issues/2353
@@ -70,7 +73,8 @@ class PyH5py(PythonPackage):
 
     # Link dependencies (py-h5py v2 cannot build against HDF5 1.12 regardless
     # of API setting)
-    depends_on("hdf5@1.10.4:1.14 +hl", when="@3.10:")
+    depends_on("hdf5@1.10.6:1.14 +hl", when="@3.12:")
+    depends_on("hdf5@1.10.4:1.14 +hl", when="@3.10:3.11")
     depends_on("hdf5@1.8.4:1.14 +hl", when="@3.8:3.9")
     depends_on("hdf5@1.8.4:1.12 +hl", when="@3:3.7")
     depends_on("hdf5@1.8.4:1.11 +hl", when="@:2")
diff --git a/var/spack/repos/builtin/packages/py-jaxtyping/package.py b/var/spack/repos/builtin/packages/py-jaxtyping/package.py
index fb703a3eaa..a64a87d158 100644
--- a/var/spack/repos/builtin/packages/py-jaxtyping/package.py
+++ b/var/spack/repos/builtin/packages/py-jaxtyping/package.py
@@ -15,7 +15,16 @@ class PyJaxtyping(PythonPackage):
     license("Apache-2.0")
 
     version("0.2.33", sha256="9a9cfccae4fe05114b9fb27a5ea5440be4971a5a075bbd0526f6dd7d2730f83e")
+    version("0.2.19", sha256="21ff4c3caec6781cadfe980b019dde856c1011e17d11dfe8589298040056325a")
 
-    depends_on("python@3.9:", type=("build", "run"))
     depends_on("py-hatchling", type="build")
-    depends_on("py-typeguard@2.13.3", type=("build", "run"))
+
+    with default_args(type=("build", "run")):
+        depends_on("python@3.9:3", when="@0.2.33:")
+        depends_on("python@3.8:3", when="@0.2.19")
+        depends_on("py-typeguard@2.13.3", when="@0.2.33:")
+        depends_on("py-typeguard@2.13.3:", when="@0.2.19")
+
+        # Historical dependencies
+        depends_on("py-numpy@1.12:", when="@0.2.19")
+        depends_on("py-typing-extensions@3.7.4.1:", when="@0.2.19")
diff --git a/var/spack/repos/builtin/packages/py-laplace-torch/package.py b/var/spack/repos/builtin/packages/py-laplace-torch/package.py
new file mode 100644
index 0000000000..82a95a462c
--- /dev/null
+++ b/var/spack/repos/builtin/packages/py-laplace-torch/package.py
@@ -0,0 +1,31 @@
+# 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 PyLaplaceTorch(PythonPackage):
+    """laplace - Laplace approximations for deep learning."""
+
+    homepage = "https://github.com/aleximmer/Laplace"
+    pypi = "laplace_torch/laplace_torch-0.2.1.tar.gz"
+
+    license("MIT")
+
+    version("0.2.1", sha256="641823a6d3e1dcb8297202b896ae2969334bf96df9a4a6f8cf688896d67d96f2")
+
+    with default_args(type="build"):
+        depends_on("py-setuptools@42:")
+        depends_on("py-setuptools-scm")
+
+    with default_args(type=("build", "run")):
+        depends_on("py-torch@2:")
+        depends_on("py-torchvision")
+        depends_on("py-torchaudio")
+        depends_on("py-backpack-for-pytorch")
+        depends_on("py-asdfghjkl@0.1a4")
+        depends_on("py-torchmetrics")
+        depends_on("py-opt-einsum")
+        depends_on("py-curvlinops-for-pytorch@2:")
diff --git a/var/spack/repos/builtin/packages/py-lightning-uq-box/package.py b/var/spack/repos/builtin/packages/py-lightning-uq-box/package.py
new file mode 100644
index 0000000000..ec4eab6acc
--- /dev/null
+++ b/var/spack/repos/builtin/packages/py-lightning-uq-box/package.py
@@ -0,0 +1,47 @@
+# 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 PyLightningUqBox(PythonPackage):
+    """Lighning-UQ-Box: A toolbox for uncertainty quantification in deep learning."""
+
+    homepage = "https://github.com/lightning-uq-box/lightning-uq-box"
+    pypi = "lightning-uq-box/lightning-uq-box-0.1.0.tar.gz"
+    git = "https://github.com/lightning-uq-box/lightning-uq-box.git"
+
+    license("Apache-2.0")
+    maintainers("nilsleh", "adamjstewart")
+
+    version("main", branch="main")
+    version("0.1.0", sha256="ce44860db75b4fbe487a009bee91c886be2e1835edee93479a6a8633ef2152b1")
+
+    depends_on("py-setuptools@61:", type="build")
+
+    with default_args(type=("build", "run")):
+        depends_on("python@3.10:", when="@0.2:")
+        depends_on("python@3.9:")
+        depends_on("py-einops@0.3:")
+        depends_on("py-lightning@2.4:", when="@0.2:")
+        depends_on("py-lightning@2.1.1:")
+        depends_on("py-matplotlib@3.5:", when="@0.2:")
+        depends_on("py-matplotlib@3.3.3:")
+        depends_on("py-numpy@1.21.1:", when="@0.2:")
+        depends_on("py-numpy@1.19.3:")
+        depends_on("py-pandas@1.1.3:")
+        depends_on("py-torch@2:")
+        depends_on("py-torchmetrics@1.2:")
+        depends_on("py-torchvision@0.16.1:")
+        depends_on("py-scikit-learn@1.3:")
+        depends_on("py-gpytorch@1.11:")
+        depends_on("py-laplace-torch@0.2.1:", when="@0.2:")
+        depends_on("py-laplace-torch@0.1:")
+        depends_on("py-uncertainty-toolbox@0.1.1:")
+        depends_on("py-kornia@0.6.9:")
+        depends_on("py-timm@0.9.2:")
+        depends_on("py-torchseg@0.0.1:")
+        depends_on("py-h5py@3.12.1:", when="@0.2:")
+        depends_on("py-ema-pytorch@0.7:", when="@0.2:")
diff --git a/var/spack/repos/builtin/packages/py-linear-operator/package.py b/var/spack/repos/builtin/packages/py-linear-operator/package.py
index 00c7899413..89034f15c7 100644
--- a/var/spack/repos/builtin/packages/py-linear-operator/package.py
+++ b/var/spack/repos/builtin/packages/py-linear-operator/package.py
@@ -17,6 +17,7 @@ class PyLinearOperator(PythonPackage):
 
     license("MIT")
 
+    version("0.5.3", sha256="16122661cd8b8a89ea965c845f650affe0f688f315893bb8dfa1182f148a1a41")
     version("0.4.0", sha256="7c57c9f8f258c9785c0db4dd7625f4dd03a340313d7314cba0b633644909f5c6")
     version("0.3.0", sha256="84bf572631a7e1576de6920d81600ca0fedcf6bda2f29dbaf440d6e72ce6abab")
     version("0.1.1", sha256="81adc1aea9e98f3c4f07f5608eb77b689bc61793e9beebfea82155e9237bf1be")
@@ -26,3 +27,5 @@ class PyLinearOperator(PythonPackage):
     depends_on("py-setuptools-scm", type="build")
     depends_on("py-torch@1.11:", type=("build", "run"))
     depends_on("py-scipy", type=("build", "run"))
+    depends_on("py-jaxtyping@0.2.19", when="@0.5.3:", type=("build", "run"))
+    depends_on("py-mpmath@0.19:1.3", when="@0.5.3:", type=("build", "run"))
diff --git a/var/spack/repos/builtin/packages/py-timm/package.py b/var/spack/repos/builtin/packages/py-timm/package.py
index 7a993364b6..dbf659b68c 100644
--- a/var/spack/repos/builtin/packages/py-timm/package.py
+++ b/var/spack/repos/builtin/packages/py-timm/package.py
@@ -12,10 +12,10 @@ class PyTimm(PythonPackage):
     homepage = "https://github.com/rwightman/pytorch-image-models"
     pypi = "timm/timm-0.4.12.tar.gz"
 
-    maintainers("adamjstewart")
-
     license("Apache-2.0")
+    maintainers("adamjstewart")
 
+    version("1.0.11", sha256="a005f72b87e67ed30cdbf405a9ffd4e723360c780a43b1cefe266af8ecc9d151")
     version("0.9.7", sha256="2bfb1029e90b72e65eb9c75556169815f2e82257eaa1f6ebd623a4b4a52867a2")
     version("0.9.5", sha256="669835f0030cfb2412c464b7b563bb240d4d41a141226afbbf1b457e4f18cff1")
     version("0.9.2", sha256="d0977cc5e02c69bda979fca8b52aa315a5f2cb64ebf8ad2c4631b1e452762c14")
@@ -26,20 +26,22 @@ class PyTimm(PythonPackage):
     version("0.5.4", sha256="5d7b92e66a76c432009aba90d515ea7a882aae573415a7c5269e3617df901c1f")
     version("0.4.12", sha256="b14be70dbd4528b5ca8657cf5bc2672c7918c3d9ebfbffe80f4785b54e884b1e")
 
-    # https://github.com/huggingface/pytorch-image-models/commit/f744bda994ec305a823483bf52a20c1440205032
-    depends_on("python@3.8:", when="@0.9.0", type=("build", "run"))
-    # https://github.com/huggingface/pytorch-image-models/issues/1530
-    # https://github.com/huggingface/pytorch-image-models/pull/1649
-    depends_on("python@:3.10", when="@:0.6.12", type=("build", "run"))
-
-    depends_on("py-setuptools", type="build")
-    depends_on("py-torch@1.7:", when="@0.6:", type=("build", "run"))
-    depends_on("py-torch@1.4:", type=("build", "run"))
-    depends_on("py-torchvision", type=("build", "run"))
-    depends_on("py-pyyaml", when="@0.6:", type=("build", "run"))
-    depends_on("py-huggingface-hub", when="@0.6:", type=("build", "run"))
-    depends_on("py-safetensors", when="@0.9:", type=("build", "run"))
-
-    # https://github.com/rwightman/pytorch-image-models/pull/1256
-    depends_on("pil@:9", when="@:0.5", type=("build", "run"))
-    depends_on("py-numpy", when="@:0.5", type=("build", "run"))
+    with default_args(type="build"):
+        depends_on("py-pdm-backend", when="@1:")
+        depends_on("py-setuptools", when="@:0")
+
+    with default_args(type=("build", "run")):
+        # https://github.com/huggingface/pytorch-image-models/issues/1530
+        # https://github.com/huggingface/pytorch-image-models/pull/1649
+        depends_on("python@:3.10", when="@:0.6.12", type=("build", "run"))
+
+        depends_on("py-torch@1.7:", when="@0.6:", type=("build", "run"))
+        depends_on("py-torch@1.4:", type=("build", "run"))
+        depends_on("py-torchvision", type=("build", "run"))
+        depends_on("py-pyyaml", when="@0.6:", type=("build", "run"))
+        depends_on("py-huggingface-hub", when="@0.6:", type=("build", "run"))
+        depends_on("py-safetensors", when="@0.9:", type=("build", "run"))
+
+        # https://github.com/rwightman/pytorch-image-models/pull/1256
+        depends_on("pil@:9", when="@:0.5", type=("build", "run"))
+        depends_on("py-numpy", when="@:0.5", type=("build", "run"))
diff --git a/var/spack/repos/builtin/packages/py-torchseg/package.py b/var/spack/repos/builtin/packages/py-torchseg/package.py
new file mode 100644
index 0000000000..e8f7a2cebc
--- /dev/null
+++ b/var/spack/repos/builtin/packages/py-torchseg/package.py
@@ -0,0 +1,26 @@
+# 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 PyTorchseg(PythonPackage):
+    """TorchSeg: Semantic Segmentation models for PyTorch."""
+
+    homepage = "https://github.com/isaaccorley/torchseg"
+    pypi = "torchseg/torchseg-0.0.1a4.tar.gz"
+
+    maintainers("isaaccorley", "adamjstewart")
+
+    license("MIT")
+
+    version("0.0.1a4", sha256="4742551753599af92f9f85e5ca6b149b474ffd458bad1aad6b3aad246a3bf4ea")
+
+    depends_on("py-setuptools@61:")
+
+    with default_args(type=("build", "run")):
+        depends_on("py-einops@0.7:")
+        depends_on("py-timm@0.9.12:")
+        depends_on("py-torch@1.13:")
diff --git a/var/spack/repos/builtin/packages/py-uncertainty-toolbox/package.py b/var/spack/repos/builtin/packages/py-uncertainty-toolbox/package.py
new file mode 100644
index 0000000000..c6b0aadbb0
--- /dev/null
+++ b/var/spack/repos/builtin/packages/py-uncertainty-toolbox/package.py
@@ -0,0 +1,29 @@
+# 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 PyUncertaintyToolbox(PythonPackage):
+    """Uncertainty Toolbox: a python toolbox for predictive uncertainty quantification,
+    calibration, metrics, and visualization."""
+
+    homepage = "https://uncertainty-toolbox.github.io/"
+    pypi = "uncertainty-toolbox/uncertainty-toolbox-0.1.1.tar.gz"
+
+    license("MIT")
+
+    version("0.1.1", sha256="d9389112bd431edc8b6e44c5b12405dea8f86063ff9b79f0bb178e5ac76bcfa5")
+
+    with default_args(type="build"):
+        depends_on("py-flit-core@3.2:3")
+        depends_on("py-setuptools")
+
+    with default_args(type=("build", "run")):
+        depends_on("py-numpy@1.19:")
+        depends_on("py-scipy@1.5:")
+        depends_on("py-matplotlib@3.2.2:")
+        depends_on("py-scikit-learn@0.23.1:")
+        depends_on("py-tqdm@4.54:")
diff --git a/var/spack/repos/builtin/packages/py-unfoldnd/package.py b/var/spack/repos/builtin/packages/py-unfoldnd/package.py
new file mode 100644
index 0000000000..e3f1fec171
--- /dev/null
+++ b/var/spack/repos/builtin/packages/py-unfoldnd/package.py
@@ -0,0 +1,25 @@
+# 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 PyUnfoldnd(PythonPackage):
+    """N-dimensional unfold (im2col) and fold (col2im) in PyTorch."""
+
+    homepage = "https://github.com/f-dangel/unfoldNd"
+    pypi = "unfoldnd/unfoldnd-0.2.2.tar.gz"
+
+    license("MIT")
+
+    version("0.2.2", sha256="e8fdffeb68bc1b393ddc1b1c87056e0e4616db992e95c7dbc3dc90d564599397")
+
+    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")
+        depends_on("py-numpy")