zhangyiss/spack
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add py-equinox (#49342)

Browse Source 
Adds Equinox, a JAX library. I've added the latest version 0.11.2, and also 0.11.0
which is compatible with older JAX versions.

I've built both versions and loading an example from their page seems to work OK.

* Add py-wadler-lindig
* Add py-equinox
This commit is contained in:
Fernando Ayats 2025-03-28 10:27:04 +01:00 committed by GitHub
parent bb1216432a
commit 03d70feb18
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 54 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
var/spack/repos/builtin/packages/py-equinox/package.py Normal file
View File

@ -0,0 +1,33 @@
# Copyright Spack Project Developers. See COPYRIGHT file for details.
#
# SPDX-License-Identifier: (Apache-2.0 OR MIT)
from spack.package import *
class PyEquinox(PythonPackage):
"""Equinox is a comprehensive JAX library that provides a wide
range of tools and features not found in core JAX, including neural networks
with PyTorch-like syntax, filtered APIs for transformations, PyTree manipulation
routines, and advanced features like runtime errors."""
homepage = "https://docs.kidger.site/equinox/"
pypi = "equinox/equinox-0.11.10.tar.gz"
maintainers("viperMl")
license("Apache-2.0", checked_by="viperML")
version("0.11.12", sha256="bee22aabaf7ee0cde6f2ae58cf3b981dea73d47e297361a0203e299208ef1739")
version("0.11.10", sha256="f3e7d5545b71e427859a28050526d09adb6b20285c47476a606328a0b96c9509")
depends_on("py-hatchling", type="build")
with default_args(type=("build", "run")):
depends_on("python@3.10:")
depends_on("py-jax@0.4.13:0.4.26", when="@:0.11.10")
depends_on("py-jax@0.4.28:", when="@0.11.11:")
depends_on("py-jaxtyping@0.2.20:")
depends_on("py-typing-extensions@4.5.0:")
depends_on("py-wadler-lindig@0.1.0:")

21
var/spack/repos/builtin/packages/py-wadler-lindig/package.py Normal file
View File

@ -0,0 +1,21 @@
# Copyright Spack Project Developers. See COPYRIGHT file for details.
#
# SPDX-License-Identifier: (Apache-2.0 OR MIT)
from spack.package import *
class PyWadlerLindig(PythonPackage):
"""A Wadler–Lindig pretty-printer for Python"""
homepage = "https://docs.kidger.site/wadler_lindig"
pypi = "wadler_lindig/wadler_lindig-0.1.3.tar.gz"
version("0.1.3", sha256="476fb7015135f714cef8f8eac7c44b164c8b993345e651a9b6f25b7b112440c9")
maintainers("viperML")
license("Apache-2.0", checked_by="viperML")
depends_on("py-hatchling", type="build")
depends_on("python@3.10:", type=("build", "run"))