Ecosyste.ms: Packages

An open API service providing package, version and dependency metadata of many open source software ecosystems and registries.

Top 5.3% on spack.io
Top 1.8% forks on spack.io

spack.io : py-jaxlib

XLA library for Jax

Registry - Source - JSON
purl: pkg:spack/py-jaxlib
Keywords: jax
License: Other
Latest release: about 2 months ago
First release: over 1 year ago
Dependent packages: 6
Stars: 22,297 on GitHub
Forks: 2,091 on GitHub
See more repository details: repos.ecosyste.ms
Last synced: 27 days ago

py-chex 0.1.85
Chex is a library of utilities for helping to write reliable JAX code.
3 versions - Latest release: about 2 months ago - 3 dependent packages - 533 stars on GitHub - 1 maintainer
py-dm-haiku 0.0.12
JAX-based neural network library
3 versions - Latest release: about 2 months ago - 2 dependent packages - 2,405 stars on GitHub - 1 maintainer
Top 3.0% on spack.io
py-jax 0.4.26
JAX is Autograd and XLA, brought together for high-performance machine learning research. With it...
26 versions - Latest release: about 2 months ago - 14 dependent packages - 22,297 stars on GitHub - 1 maintainer
py-optax 0.2.1
A gradient processing and optimisation library in JAX.
2 versions - Latest release: 2 months ago - 1 dependent package - 1,290 stars on GitHub - 1 maintainer
py-orbax-checkpoint 0.5.3
Orbax includes a checkpointing library oriented towards JAX users, supporting a variety of differ...
1 version - Latest release: 3 months ago - 1 dependent package - 206 stars on GitHub
py-jraph 0.0.6.dev0
Jraph: A library for Graph Neural Networks in Jax.
1 version - Latest release: 3 months ago - 1 dependent package - 1,292 stars on GitHub