PyPI
nstarman/quax
Multiple dispatch over abstract array types in JAX.
Apache-2.0Oct 4, 2026 · metrics 2.10.0
All tags
Catalogue tag
#equinox
Every repository in the public record carrying this tag — from its GitHub topics or the keywords its package registries publish. Health is measured under the same versioned methodology as the rest of the record.
8 records
PyPI
patrick-kidger/equinox
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
Apache-2.0Sep 14, 2026 · metrics 2.10.0
PyPI
patrick-kidger/jaxtyping
Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/
Custom licenseAug 27, 2026 · metrics 2.10.0
PyPI
patrick-kidger/diffrax
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
Apache-2.0Sep 14, 2026 · metrics 2.10.0
PyPI
patrick-kidger/lineax
Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax
Apache-2.0Sep 17, 2026 · metrics 2.10.0
PyPI
patrick-kidger/optimistix
Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/
Apache-2.0Sep 17, 2026 · metrics 2.10.0
PyPI
danielward27/flowjax
No repository description published.
MITOct 16, 2026 · metrics 2.10.0
PyPI
42Ar/jaxon
Save pytrees efficiently in hdf5 files.
GPL-3.0Sep 17, 2026 · metrics 2.10.0