Ecosyste.ms: Packages

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

pypi.org "jax" keyword

google-fax 0.0.2
FAX is an implementation of federated operations using Jax primitives.
2 versions - Latest release: 3 months ago - 7 downloads last month - 32,863 stars on GitHub - 1 maintainer
jaxutils 0.0.8
Utility functions for JaxGaussianProcesses
8 versions - Latest release: over 1 year ago - 4 dependent packages - 1 dependent repositories - 304 downloads last month - 2 maintainers
jaxkern-nightly 0.0.5.dev20240517 💰
Kernels in Jax.
524 versions - Latest release: 16 days ago - 3.47 thousand downloads last month - 7 stars on GitHub - 1 maintainer
jaxutils-nightly 0.0.8.dev20240517
Utility functions for JaxGaussianProcesses
441 versions - Latest release: 16 days ago - 2.74 thousand downloads last month - 1 maintainer
ripplegw 0.0.7
A small jax package for differentiable and fast gravitational wave data analysis.
9 versions - Latest release: 23 days ago - 1 dependent package - 532 downloads last month - 48 stars on GitHub - 1 maintainer
Top 0.3% on pypi.org
flax 0.8.3
Flax: A neural network library for JAX designed for flexibility
43 versions - Latest release: about 1 month ago - 198 dependent packages - 3,260 dependent repositories - 4.22 million downloads last month - 5,583 stars on GitHub - 5 maintainers
Top 1.5% on pypi.org
numpyro 0.15.0
Pyro PPL on NumPy
28 versions - Latest release: 19 days ago - 76 dependent packages - 123 dependent repositories - 146 thousand downloads last month - 2,057 stars on GitHub - 1 maintainer
invrs-gym 0.10.3
A collection of inverse design challenges
26 versions - Latest release: about 2 months ago - 1 dependent package - 300 downloads last month - 87 stars on GitHub - 1 maintainer
fmmax 0.8.0
Fourier modal method with Jax
16 versions - Latest release: 2 months ago - 1 dependent package - 1 dependent repositories - 441 downloads last month - 87 stars on GitHub - 1 maintainer
jax2torch 0.0.7
Jax 2 Torch
7 versions - Latest release: over 2 years ago - 1 dependent repositories - 84 downloads last month - 212 stars on GitHub - 1 maintainer
meent 0.9.10
Electromagnetic simulation (RCWA) & optimization package in Python
47 versions - Latest release: about 1 month ago - 1 dependent repositories - 116 downloads last month - 40 stars on GitHub - 1 maintainer
coordinax 0.4.3
Vectors in JAX
9 versions - Latest release: 2 months ago - 42 downloads last month - 3 stars on GitHub - 2 maintainers
jvt 9.7.22
Vision transformers with JAX & Flax
2 versions - Latest release: over 1 year ago - 14 downloads last month - 4 stars on GitHub - 1 maintainer
wax-ml 0.6.4
A Python library for machine-learning and feedback loops on streaming data
13 versions - Latest release: about 1 year ago - 1 dependent repositories - 28 downloads last month - 57 stars on GitHub - 1 maintainer
veros 1.5.1
The versatile ocean simulator, in pure Python, powered by JAX.
19 versions - Latest release: 8 months ago - 4 dependent repositories - 360 downloads last month - 298 stars on GitHub - 1 maintainer
sunstreaker 0.0.1.dev5
源码清晰明了,使用简单好搞
5 versions - Latest release: over 1 year ago - 16 downloads last month - 123 stars on GitHub - 1 maintainer
Top 2.9% on pypi.org
pyhf 0.7.6
pure-Python HistFactory implementation with tensors and autodiff
47 versions - Latest release: 5 months ago - 5 dependent packages - 27 dependent repositories - 5.3 thousand downloads last month - 274 stars on GitHub - 4 maintainers
hal9 2.1.0
🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch and FLAX.
38 versions - Latest release: 15 days ago - 19.8 thousand downloads last month - 22,366 stars on GitHub - 2 maintainers
soujpg-diffusers 0.27.2
State-of-the-art diffusion in PyTorch and JAX.
6 versions - Latest release: 2 months ago - 89 downloads last month - 22,366 stars on GitHub - 1 maintainer
rationalnets 0.1.0
JAX/Flax implementation of rational neural nets
2 versions - Latest release: over 1 year ago - 1 dependent repositories - 26 downloads last month - 1 stars on GitHub - 1 maintainer
mosec-tiinfer 0.0.7
Model Serving made Efficient in the Cloud.
1 version - Latest release: about 1 year ago - 18 downloads last month - 712 stars on GitHub - 1 maintainer
Top 7.4% on pypi.org
mosec 0.8.4
Model Serving made Efficient in the Cloud
42 versions - Latest release: 3 months ago - 1 dependent package - 1 dependent repositories - 1.59 thousand downloads last month - 657 stars on GitHub - 2 maintainers
jimm 0.0.1
JAX Image Models
1 version - Latest release: over 3 years ago - 1 dependent repositories - 14 downloads last month - 1 maintainer
paxml 1.4.0
Framework to configure and run machine learning experiments on top of Jax.
12 versions - Latest release: about 2 months ago - 9.25 thousand downloads last month - 414 stars on GitHub - 1 maintainer
xpag 0.2.9
xpag: Exploring Agents
18 versions - Latest release: 7 months ago - 36 downloads last month - 20 stars on GitHub - 1 maintainer
jaxnrsur 0.0.1
Numerical relativity surrogate model for gravitational waveforms in Jax
1 version - Latest release: 28 days ago - 235 downloads last month - 4 stars on GitHub - 1 maintainer
Top 1.5% on pypi.org
pennylane 0.36.0
PennyLane is a cross-platform Python library for quantum computing, quantum machine learning, and...
50 versions - Latest release: 26 days ago - 49 dependent packages - 95 dependent repositories - 52.4 thousand downloads last month - 2,101 stars on GitHub - 6 maintainers
Top 1.3% on pypi.org
orbax-checkpoint 0.5.11
Orbax Checkpoint
45 versions - Latest release: 22 days ago - 16 dependent packages - 313 dependent repositories - 21.1 million downloads last month - 221 stars on GitHub - 1 maintainer
iqa-jax 0.0.1
IQA library for Jax
1 version - Latest release: about 1 year ago - 19 downloads last month - 0 stars on GitHub - 1 maintainer
Top 0.7% on pypi.org
pytorch-transformers 1.2.0
Repository of pre-trained NLP Transformer models: BERT & RoBERTa, GPT & GPT-2, Transformer-XL, XL...
4 versions - Latest release: over 4 years ago - 16 dependent packages - 772 dependent repositories - 40.2 thousand downloads last month - 123,549 stars on GitHub - 1 maintainer
google-jetstream 0.2.1
JetStream is a throughput and memory optimized engine for LLM inference on XLA devices, starting ...
5 versions - Latest release: 29 days ago - 11.6 thousand downloads last month - 126 stars on GitHub - 1 maintainer
ngclearn 1.0b4
Simulation software for building and analyzing arbitrary predictive coding, spiking network, and ...
5 versions - Latest release: about 1 month ago - 110 downloads last month - 85 stars on GitHub - 1 maintainer
fjformer 0.0.57
Embark on a journey of paralleled/unparalleled computational prowess with FJFormer - an arsenal o...
52 versions - Latest release: 18 days ago - 1 dependent package - 1.27 thousand downloads last month - 21 stars on GitHub - 1 maintainer
kimm 0.1.8
A Keras model zoo with pretrained weights.
11 versions - Latest release: 2 months ago - 360 downloads last month - 27 stars on GitHub - 1 maintainer
Top 0.5% on pypi.org
etils 1.8.0
Collection of common python utils
27 versions - Latest release: 2 months ago - 22 dependent packages - 567 dependent repositories - 3.68 million downloads last month - 171 stars on GitHub - 1 maintainer
Top 6.0% on pypi.org
gpjax 0.8.2 💰
Gaussian processes in JAX.
58 versions - Latest release: 2 months ago - 6 dependent packages - 2 dependent repositories - 945 downloads last month - 387 stars on GitHub - 2 maintainers
swanlab 0.3.2
Python library for streamlined tracking and management of AI training processes.
21 versions - Latest release: 16 days ago - 777 downloads last month - 182 stars on GitHub - 1 maintainer
jaxid
A code base for system identification with Jax.
1 version - 123 downloads last month - 3 stars on GitHub - 1 maintainer
mubelnet 0.0.8
Deep Bayesian unsupervised decoder networks. Use poisson or multinomial belief networks to cluste...
1 version - Latest release: 6 months ago - 12 downloads last month - 0 stars on GitLab.com - 1 maintainer
Top 0.1% on pypi.org
keras 3.3.3
Multi-backend Keras.
99 versions - Latest release: about 1 month ago - 587 dependent packages - 24,558 dependent repositories - 18.2 million downloads last month - 60,873 stars on GitHub - 2 maintainers
pcgym 0.1.6
Reinforcement learning suite of process control problems.
11 versions - Latest release: about 2 months ago - 47 downloads last month - 16 stars on GitHub - 1 maintainer
Top 0.2% on pypi.org
jax 0.4.28
Differentiate, compile, and transform Numpy code.
162 versions - Latest release: 23 days ago - 785 dependent packages - 6,336 dependent repositories - 6.36 million downloads last month - 28,003 stars on GitHub - 9 maintainers
Top 0.2% on pypi.org
jaxlib 0.4.28
XLA library for JAX
55 versions - Latest release: 23 days ago - 511 dependent packages - 4,102 dependent repositories - 4.85 million downloads last month - 28,003 stars on GitHub - 9 maintainers
segnn-jax 0.7
Steerable E(3) GNN in jax
5 versions - Latest release: 11 months ago - 15 downloads last month - 21 stars on GitHub - 1 maintainer
rockpool 2.5.1
A Python package for developing, simulating and training spiking neural networks, and deploying o...
27 versions - Latest release: over 1 year ago - 1 dependent repositories - 214 downloads last month - 44 stars on GitHub - 2 maintainers
gradients 0.0.3
Gradient Checker for Custom built PyTorch Models
3 versions - Latest release: over 2 years ago - 1 dependent repositories - 42 downloads last month - 0 stars on GitHub - 1 maintainer
vit-flax 0.0.3
Vision Transformer, ViT - Flax
3 versions - Latest release: over 3 years ago - 1 dependent repositories - 10 downloads last month - 13 stars on GitHub - 1 maintainer
dopamax 0.2.0
Reinforcement learning in pure JAX.
2 versions - Latest release: 5 months ago - 15 downloads last month - 10 stars on GitHub - 1 maintainer
jit-env 0.1.5
A Jax interface for Reinforcement Learning environments.
8 versions - Latest release: 7 months ago - 50 downloads last month - 9 stars on GitHub - 1 maintainer
jimgw 0.1.1
Gravitatioanl wave data analysis tool in Jax
4 versions - Latest release: 9 months ago - 50 downloads last month - 49 stars on GitHub - 1 maintainer
jaxfss 0.1.0
JAX/Flax implementation of finite-size scaling
2 versions - Latest release: over 1 year ago - 31 downloads last month - 8 stars on GitHub - 1 maintainer
Top 0.1% on pypi.org
transformers 4.40.2
State-of-the-art Machine Learning for JAX, PyTorch and TensorFlow
144 versions - Latest release: 27 days ago - 2,589 dependent packages - 31,800 dependent repositories - 28.1 million downloads last month - 126,450 stars on GitHub - 4 maintainers
Top 3.8% on pypi.org
kubeflow-katib 0.16.0
Katib Python SDK for APIVersion v1beta1
24 versions - Latest release: 7 months ago - 59 dependent repositories - 3.37 thousand downloads last month - 1,419 stars on GitHub - 6 maintainers
sequel-core 0.0.2
A Continual Learning Framework for both Jax and PyTorch.
2 versions - Latest release: about 1 year ago - 14 downloads last month - 0 stars on GitHub - 1 maintainer
redco 0.4.16
NAACL '24 (Demo) / MlSys @ NeurIPS '23 - RedCoast: A Lightweight Tool to Automate Distributed Tra...
17 versions - Latest release: about 1 month ago - 46 downloads last month - 41 stars on GitHub - 1 maintainer
gpjax-nightly 0.5.8 💰
Didactic Gaussian processes in Jax.
56 versions - Latest release: over 1 year ago - 187 downloads last month - 387 stars on GitHub - 1 maintainer
ref-gsd 0.2.3
Reference implementation of generalised score distribution in python
9 versions - Latest release: 3 months ago - 55 downloads last month - 1 stars on GitHub - 1 maintainer
Top 0.6% on pypi.org
tensorflow-datasets 4.9.4
tensorflow/datasets is a library of datasets ready to use with TensorFlow.
34 versions - Latest release: 6 months ago - 116 dependent packages - 3,946 dependent repositories - 4.02 million downloads last month - 4,085 stars on GitHub - 8 maintainers
Top 6.4% on pypi.org
netket 3.12.0
Netket : Machine Learning toolbox for many-body quantum systems.
63 versions - Latest release: 19 days ago - 5 dependent repositories - 1.66 thousand downloads last month - 498 stars on GitHub - 2 maintainers
pychastic 0.2.1
Solvers for stochastic differential equations
6 versions - Latest release: about 1 year ago - 1 dependent repositories - 23 downloads last month - 8 stars on GitHub - 1 maintainer
memory-efficient-attention 0.1.3
Memory Efficient Attention (O(sqrt(n)) for Jax and PyTorch
4 versions - Latest release: about 2 years ago - 1 dependent repositories - 43 downloads last month - 175 stars on GitHub - 1 maintainer
cryojax 0.2.3
Cryo-EM image simulation and analysis powered by JAX
7 versions - Latest release: about 2 months ago - 261 downloads last month - 26 stars on GitHub - 1 maintainer
quaxed 0.4.2
Array-API JAX compatibility
13 versions - Latest release: 25 days ago - 3 dependent packages - 3.8 thousand downloads last month - 2 stars on GitHub - 2 maintainers
Top 1.8% on pypi.org
equinox 0.11.4 💰
Elegant easy-to-use neural networks in JAX.
48 versions - Latest release: about 2 months ago - 78 dependent packages - 26 dependent repositories - 260 thousand downloads last month - 1,846 stars on GitHub - 1 maintainer
jaxtronomy 0.0.1rc1
lenstronomy, but in JAX
1 version - Latest release: 3 months ago - 6 downloads last month - 14 stars on GitHub - 1 maintainer
jammer 0.1
Library for jax based affine-invariant MCMC sampling
1 version - Latest release: about 3 years ago - 1 dependent repositories - 18 downloads last month - 0 stars on GitHub - 1 maintainer
diffeqzoo 0.0.5
Ordinary differential equation zoo
5 versions - Latest release: 5 months ago - 1 dependent package - 173 downloads last month - 8 stars on GitHub - 1 maintainer
last-asr 0.0.4
The LAttice-based Speech Transducer (LAST) library
4 versions - Latest release: 2 months ago - 25 downloads last month - 37 stars on GitHub - 1 maintainer
thermox 0.0.0
OU Processes and Linear Algebra with JAX
2 versions - Latest release: 3 months ago - 15 downloads last month - 2 maintainers
levy-stable-jax 0.2.0
Implementation of the Lévy alpha-stable distributions for Jax.
6 versions - Latest release: 25 days ago - 647 downloads last month - 0 stars on GitHub - 1 maintainer
differt-core 0.0.11 💰
Core backend of DifffeRT implemented in Rust
8 versions - Latest release: 30 days ago - 2 dependent packages - 4.18 thousand downloads last month - 1 stars on GitHub - 1 maintainer
differt 0.0.10 💰
Differentiable Ray Tracing Toolbox for Radio Propagation Simulations
12 versions - Latest release: 3 months ago - 2.74 thousand downloads last month - 1 stars on GitHub - 1 maintainer
discco 0.2.7
Quantification of membrane and cytoplasmic concentrations based on differentiable simulation of c...
9 versions - Latest release: 7 months ago - 21 downloads last month - 0 stars on GitHub - 1 maintainer
ezjaxtyping 0.2.20
Type annotations and runtime checking for shape and dtype of JAX arrays, and PyTrees.
1 version - Latest release: 11 months ago - 1 dependent package - 21 downloads last month - 0 stars on GitHub - 1 maintainer
nebula-loss 0.4.1 💰
1 Loss Function to rule them all!
5 versions - Latest release: 10 months ago - 29 downloads last month - 5 stars on GitHub - 1 maintainer
jaxtro 0.0.2
A JAX-based gravitational-wave population inference
2 versions - Latest release: 5 months ago - 13 downloads last month - 5 stars on GitHub - 1 maintainer
pygrpy 0.1.3
Python port of Generalized Rotne Prager Yakamava tensors
11 versions - Latest release: 5 months ago - 2 dependent packages - 1 dependent repositories - 87 downloads last month - 1 stars on GitHub - 1 maintainer
holab 0.0.3
Acoustic lens design on GPUs with Automatic Differentiation.
2 versions - Latest release: 9 months ago - 19 downloads last month - 4 stars on GitHub - 1 maintainer
pynajax 0.1.2
A JAX backend for pynapple.
3 versions - Latest release: 19 days ago - 208 downloads last month - 6 stars on GitHub - 1 maintainer
penzai 0.1.1
Penzai: A JAX research toolkit for building, editing, and visualizing neural networks.
3 versions - Latest release: 29 days ago - 1 dependent package - 3.8 thousand downloads last month - 1,445 stars on GitHub - 2 maintainers
teneva-jax 0.1.1
Compact implementation of basic operations in the tensor-train (TT) format with jax framework for...
1 version - Latest release: about 1 year ago - 43 downloads last month - 4 stars on GitHub - 1 maintainer
jaims 0.0.1
Library for jax based affine-invariant MCMC sampling
1 version - Latest release: about 3 years ago - 1 dependent repositories - 13 downloads last month - 0 stars on GitHub - 1 maintainer
hydrax 0.2.2
A zero-copy multiprocess dataloader for JAX.
3 versions - Latest release: 2 months ago - 17 downloads last month - 1 stars on GitHub - 1 maintainer
xspex 0.0.3
Access Xspec models and corresponding JAX/XLA ops.
4 versions - Latest release: about 1 month ago - 1 dependent package - 56 downloads last month - 1 stars on GitHub - 1 maintainer
jaxsim 0.2.0
A differentiable physics engine and multibody dynamics library for control and robot learning.
85 versions - Latest release: about 2 months ago - 970 downloads last month - 50 stars on GitHub - 3 maintainers
Top 9.2% on pypi.org
tf2jax 0.3.6
TF2JAX: Convert TensorFlow to JAX
10 versions - Latest release: 2 months ago - 1 dependent package - 1 dependent repositories - 3.23 thousand downloads last month - 96 stars on GitHub - 1 maintainer
jaxoplanet 0.0.2
Probabilistic models for exoplanet inference in JAX
3 versions - Latest release: 3 months ago - 1 dependent package - 574 downloads last month - 18 stars on GitHub - 1 maintainer
portrait
Compute and visualize observations phase coverage
4 versions - 256 downloads last month - 0 stars on GitHub - 1 maintainer
jaxampler 0.0.7
A JAX-based statistical sampling toolkit
7 versions - Latest release: 4 months ago - 1 dependent package - 32 downloads last month - 9 stars on GitHub - 1 maintainer
Top 1.4% on pypi.org
dm-haiku 0.0.12
Haiku is a library for building neural networks in JAX.
15 versions - Latest release: 3 months ago - 70 dependent packages - 949 dependent repositories - 251 thousand downloads last month - 2,690 stars on GitHub - 2 maintainers
Top 3.2% on pypi.org
thinc-gpu-ops 0.0.4
CUDA kernels for Thinc
4 versions - Latest release: over 5 years ago - 2 dependent packages - 10 dependent repositories - 1.63 thousand downloads last month - 2,771 stars on GitHub - 3 maintainers
hamux 0.1.1
A Deep Learning framework built around ENERGY
2 versions - Latest release: over 1 year ago - 15 downloads last month - 82 stars on GitHub - 1 maintainer
haiku-geometric 0.0.5
A collection of graph neural networks implementations in JAX
7 versions - Latest release: 8 months ago - 1 dependent repositories - 48 downloads last month - 28 stars on GitHub - 1 maintainer
Top 0.9% on pypi.org
thinc 9.0.0
A refreshing functional take on deep learning, compatible with your favorite libraries
232 versions - Latest release: about 1 month ago - 40 dependent packages - 8,632 dependent repositories - 9.77 million downloads last month - 2,798 stars on GitHub - 3 maintainers
Top 1.4% on pypi.org
jaxtyping 0.2.28 💰
Type annotations and runtime checking for shape and dtype of JAX arrays, and PyTrees.
33 versions - Latest release: 3 months ago - 125 dependent packages - 63 dependent repositories - 595 thousand downloads last month - 974 stars on GitHub - 1 maintainer
Top 0.4% on pypi.org
diffusers 0.27.2
State-of-the-art diffusion in PyTorch and JAX.
75 versions - Latest release: 2 months ago - 187 dependent packages - 5,162 dependent repositories - 2.37 million downloads last month - 20,155 stars on GitHub - 4 maintainers
sibase 0.2.0
A basic python package that converts numerical strings with units to base units
3 versions - Latest release: almost 2 years ago - 1 dependent repositories - 15 downloads last month - 2 stars on GitHub - 1 maintainer
Top 1.8% on pypi.org
chex 0.1.86
Chex: Testing made fun, in JAX!
24 versions - Latest release: 2 months ago - 97 dependent packages - 1,399 dependent repositories - 9.23 million downloads last month - 723 stars on GitHub - 3 maintainers
invrs-utils 0.8.3
Miscellaneous utilities for the invrs-io ecosystem
17 versions - Latest release: 2 months ago - 1 dependent package - 306 downloads last month - 1 maintainer
ymdl 0.2.6
A deep learning package manager.
5 versions - Latest release: almost 2 years ago - 1 dependent repositories - 38 downloads last month - 9 stars on GitHub - 1 maintainer