Ecosyste.ms: Packages

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

spack.io "jax" keyword

py-transformers 4.35.2
State-of-the-art Natural Language Processing for TensorFlow 2.0 and PyTorch
5 versions - Latest release: 5 months ago - 1 dependent package - 86,365 stars on GitHub - 1 maintainer
py-flax 0.8.1
Flax: A neural network library for JAX designed for flexibility.
2 versions - Latest release: about 2 months ago - 5,303 stars on GitHub
Top 4.8% on spack.io
py-einops 0.8.0
Flexible and powerful tensor operations for readable and reliable code. Supports numpy, pytorch, ...
6 versions - Latest release: 4 days ago - 5 dependent packages - 6,483 stars on GitHub - 1 maintainer
py-tensorly 0.8.1
TensorLy is a Python library that aims at making tensor learning simple and accessible. It allows...
1 version - Latest release: 8 months ago - 1,429 stars on GitHub - 1 maintainer
Top 9.6% on spack.io
py-thinc 7.4.1
Thinc: Practical Machine Learning for NLP in Python.
2 versions - Latest release: about 2 years ago - 1 dependent package - 2,681 stars on GitHub - 1 maintainer
py-tree-math 0.2.0
Mathematical operations for JAX pytrees.
1 version - Latest release: about 2 months ago - 159 stars on GitHub
py-etils 1.7.0
etils (eclectic utils) is an open-source collection of utils for python.
2 versions - Latest release: about 2 months ago - 1 dependent package - 106 stars on GitHub - 1 maintainer
Top 5.3% on spack.io
py-jaxlib 0.4.26
XLA library for Jax
16 versions - Latest release: 29 days ago - 2 dependent packages - 22,297 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: 29 days ago - 7 dependent packages - 22,297 stars on GitHub - 1 maintainer
py-netket 2.1.1
NetKet is an open-source project, delivering cutting-edge methods for the study of many-body qua...
7 versions - Latest release: about 2 years ago - 422 stars on GitHub - 1 maintainer
py-tensorflow-datasets 4.4.0
tensorflow/datasets is a library of datasets ready to use with TensorFlow.
1 version - Latest release: about 2 years ago - 3,784 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: about 2 months ago - 206 stars on GitHub
py-jraph 0.0.6.dev0
Jraph: A library for Graph Neural Networks in Jax.
1 version - Latest release: about 2 months ago - 1,292 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: about 1 year ago - 263 stars on GitHub - 1 maintainer
py-pennylane 0.28.0
PennyLane is a Python quantum machine learning library by Xanadu Inc.
1 version - Latest release: over 1 year ago - 1,703 stars on GitHub - 3 maintainers
py-dm-haiku 0.0.12
JAX-based neural network library
3 versions - Latest release: 27 days ago - 1 dependent package - 2,405 stars on GitHub - 1 maintainer
py-autoray 0.5.3
Write backend agnostic numeric code compatible with any numpy-ish array library.
1 version - Latest release: over 1 year ago - 1 dependent package - 66 stars on GitHub - 1 maintainer