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-vbjax/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 PyVbjax(PythonPackage):
+    """Virtual brain models in JAX."""
+
+    homepage = "https://github.com/ins-amu/vbjax"
+    pypi = "vbjax/vbjax-0.0.16.tar.gz"
+    git = "https://github.com/ins-amu/vbjax"
+
+    # 'joblib.test' requires 'pytest'. Leave out of 'import_modules' to avoid
+    # unnecessary dependencies.
+    skip_modules = ["joblib.test"]
+
+    license("Apache-2.0")
+
+    version("0.0.16")
+
+    depends_on('py-jax', type=('build', 'link', 'run'))
+    depends_on("python", type=("build", "run"))
+    depends_on("py-setuptools", type=("build", "run"))