Ecosyste.ms: Packages
An open API service providing package, version and dependency metadata of many open source software ecosystems and registries.
Top 9.7% dependent packages on spack.io
Top 1.8% forks on spack.io
spack.io : py-jax
JAX is Autograd and XLA, brought together for high-performance machine learning research. With its updated version of Autograd, JAX can automatically differentiate native Python and NumPy functions. It can differentiate through loops, branches, recursion, and closures, and it can take derivatives of derivatives of derivatives. It supports reverse- mode differentiation (a.k.a. backpropagation) via grad as well as forward-mode differentiation, and the two can be composed arbitrarily to any order.
Registry
-
Source
- JSON
purl: pkg:spack/py-jax
Keywords: jax
License: Other
Latest release: about 2 months ago
First release: over 1 year ago
Dependent packages: 14
Stars: 22,297 on GitHub
Forks: 2,091 on GitHub
See more repository details: repos.ecosyste.ms
Last synced: 27 days ago
py-keras 3.3.2
Multi-backend Keras. Keras 3 is a new multi-backend implementation of the Keras API, with support...56 versions - Latest release: about 1 month ago - 6 dependent packages - 1 maintainer
py-tensorflow 2.16.1
TensorFlow is an open source machine learning framework for everyone.103 versions - Latest release: about 2 months ago - 13 dependent packages - 2 maintainers
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 library3 versions - Latest release: about 2 months ago - 2 dependent packages - 2,405 stars on GitHub - 1 maintainer
py-tensorflow-probability 0.23.0
TensorFlow Probability (TFP) is a Python library built on TensorFlow that makes it easy to combin...9 versions - Latest release: 2 months ago - 1 dependent package - 2 maintainers
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-tree-math 0.2.0
Mathematical operations for JAX pytrees.1 version - Latest release: 3 months ago - 159 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
py-graphcast 0.1
GraphCast: Learning skillful medium-range global weather forecasting.1 version - Latest release: 3 months ago - 4,099 stars on GitHub
py-flax 0.8.1
Flax: A neural network library for JAX designed for flexibility.2 versions - Latest release: 3 months ago - 1 dependent package - 5,303 stars on GitHub
py-mpi4jax 0.3.11.post3
Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python.1 version - Latest release: over 1 year ago - 263 stars on GitHub - 1 maintainer
py-alphafold 2.2.4
AlphaFold is an AI system developed by DeepMind that predicts a protein's 3D structure from its a...2 versions - Latest release: over 1 year ago - 10,020 stars on GitHub - 1 maintainer
py-jmp 0.0.2
JMP is a Mixed Precision library for JAX.1 version - Latest release: over 1 year ago - 1 dependent package - 123 stars on GitHub - 1 maintainer