98162aa2e1
* py-earth2mip: add new package * Fix audit tests * Make things easier to concretize * Make things easier to concretize * Fix magics build * Fix onnx build * blacken * cmake > py-cmake * Fix onnxruntime build * onnxruntime: remove unknown cmake vars * Trick bazelisk into using Spack bazel * Different deps for main * protobuf: add v3.25.3 * Enforce upper bound on jaxlib version * py-chex: add v0.1.85 * py-dm-haiku: add v0.0.12 * py-jax: add v0.4.25 * Older jaxlib doesn't build on ppc64le either
25 lines
775 B
Python
25 lines
775 B
Python
# 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 PyJraph(PythonPackage):
|
|
"""Jraph: A library for Graph Neural Networks in Jax."""
|
|
|
|
homepage = "https://github.com/deepmind/jraph"
|
|
pypi = "jraph/jraph-0.0.6.dev0.tar.gz"
|
|
|
|
license("Apache-2.0")
|
|
|
|
version(
|
|
"0.0.6.dev0", sha256="c3ac3a0b224b344eb6d367e8bc312d95ea41bf825d01ea31b80dd8c22c0dd8b8"
|
|
)
|
|
|
|
depends_on("py-setuptools", type="build")
|
|
depends_on("py-jax@0.1.55:", type=("build", "run"))
|
|
depends_on("py-jaxlib@0.1.37:", type=("build", "run"))
|
|
depends_on("py-numpy@1.18:", type=("build", "run"))
|