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 1 month 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: 12 days ago