wf-inversion

Workflow for the followings spack packages:

• py-numpyro
• py-arviz
• py-graphviz
• py-vbjax

packages/py-jaxopt/package.py

+# 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 PyJaxopt(PythonPackage):
+    """Hardware accelerated, batchable and differentiable optimizers in JAX."""
+
+    homepage = "https://github.com/google/jaxopt"
+    pypi     = "jaxopt/jaxopt-0.8.5.tar.gz"
+
+    maintainers = ['jaxopt']
+
+    version('0.8.5', sha256='ff221d1a86908ec759eb1e219ee1d12bf208a70707e961bf7401076fe7cf4d5e')
+    version('0.8.4', sha256='420878c37650c8c0752f8a75b20afb0ddc52bc12b63667910131057295b0e6ab')
+    version('0.8.3', sha256='4be2f82798393682529c9ca5046e5397ac6c8657b8acb6bf275e773b28df15b6')
+    version('0.8.2', sha256='5dd94a635ae52899d4a5063ec88b4c1ca9f04d921b888f73211a444c937b7cfa')
+    version('0.8.1', sha256='af6ef03d11d72dcf51b606d3046ebec8be0227f5d4e653cd1768cc8dab12cabb')
+
+    depends_on('python@3.7:', type=('build', 'run'))
+
+    depends_on('py-setuptools', type='build')
+
+    depends_on('py-numpy@1.18.4:', type=('build', 'run'))
+    depends_on('py-scipy@1.0.0:', type=('build', 'run'))
+    depends_on('py-jax@0.2.18:', type=('build', 'run'))
+    depends_on('py-jaxlib@0.1.69:', type=('build', 'run'))