feat(py-vbjax): add package

This adds a small JAX-based package from partner AMU vbjax which includes models developed as part of Ebrains 2.0 WP3

packages/py-numpyro/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 PyNumpyro(PythonPackage):
+    """Pyro PPL on NumPy"""
+
+    homepage = "https://github.com/pyro-ppl/numpyro"
+    pypi = "numpyro/numpyro-0.18.0.tar.gz"
+
+    version("0.18.0", sha256="9799abeb801f940b8257a7dcfab0542a2a527e0f27c34e7fc57e7c30dd113527")
+    version("0.17.0", sha256="b4ad89e3bea8280980bbae3f712bf218314869db24cd1a2e07d90cee2f85b143")
+    version("0.16.1", sha256="553ea56c729bdeb83d6eb6d455911113dc75f5eb25a59e3af31a7726313eee38")
+    version("0.16.0", sha256="9e68c26332752cc950caa91ac084c9c9bea2f60982b6dfca6acb14247540e681")
+
+    depends_on("python@3.9:", type=("build", "run"))
+    depends_on("py-setuptools", type="build")
+    depends_on("py-jax@0.4.25:", type=("build", "run"))
+    depends_on("py-jaxlib@0.4.25:", type=("build", "run"))
+    depends_on("py-multipledispatch", type=("build", "run"))
+    depends_on("py-numpy", type=("build", "run"))
+    depends_on("py-tqdm", type=("build", "run"))