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

Top 2.9% on pypi.org
pyhf 0.7.6
pure-Python HistFactory implementation with tensors and autodiff
47 versions - Latest release: 4 months ago - 5 dependent packages - 27 dependent repositories - 6.27 thousand downloads last month - 272 stars on GitHub - 4 maintainers
dks 0.1.2
A Python library implementing the DKS/TAT neural network transformation method.
3 versions - Latest release: over 1 year ago - 1 dependent repositories - 45 downloads last month - 52 stars on GitHub - 1 maintainer
dataset-grouper 0.3.0
Dataset Grouper - A library for datasets with group-level structure.
3 versions - Latest release: about 1 month ago - 38 downloads last month - 21 stars on GitHub - 6 maintainers
array-api-jax-compat 0.1.2
Array-API JAX compatibility
5 versions - Latest release: 2 months ago - 47 downloads last month - 2 stars on GitHub - 2 maintainers
Top 4.3% on pypi.org
autoray 0.6.9
Abstract your array operations.
23 versions - Latest release: 2 months ago - 23 dependent packages - 65 dependent repositories - 220 thousand downloads last month - 115 stars on GitHub - 2 maintainers
dreamerv3 1.5.0
Mastering Diverse Domains through World Models
15 versions - Latest release: about 1 year ago - 1 dependent repositories - 116 downloads last month - 847 stars on GitHub - 2 maintainers
nanodl 1.2.2.dev1 💰
A Jax-based library for designing and training transformer models from scratch.
5 versions - Latest release: about 1 month ago - 48 downloads last month - 256 stars on GitHub - 2 maintainers
Top 8.7% on pypi.org
ivy-vision 1.1.9
3D Vision functions with end-to-end support for machine learning developers, written in Ivy.
11 versions - Latest release: over 2 years ago - 3 dependent packages - 4 dependent repositories - 167 downloads last month - 71 stars on GitHub - 2 maintainers
jaxkern-nightly 0.0.5.dev20240510 💰
Kernels in Jax.
501 versions - Latest release: about 7 hours ago - 4.29 thousand downloads last month - 7 stars on GitHub - 1 maintainer
jaxutils-nightly 0.0.8.dev20240510
Utility functions for JaxGaussianProcesses
418 versions - Latest release: about 7 hours ago - 4.33 thousand downloads last month - 2 maintainers
Top 0.2% on pypi.org
jaxlib 0.4.28
XLA library for JAX
55 versions - Latest release: about 8 hours ago - 392 dependent packages - 4,102 dependent repositories - 4.35 million downloads last month - 28,003 stars on GitHub - 10 maintainers
veros 1.5.1
The versatile ocean simulator, in pure Python, powered by JAX.
19 versions - Latest release: 7 months ago - 4 dependent repositories - 272 downloads last month - 298 stars on GitHub - 2 maintainers
ripplegw 0.0.7
A small jax package for differentiable and fast gravitational wave data analysis.
7 versions - Latest release: about 10 hours ago - 1 dependent package - 229 downloads last month - 47 stars on GitHub - 2 maintainers
jax-cuda12-pjrt 0.4.28
JAX XLA PJRT Plugin for NVIDIA GPUs
7 versions - Latest release: about 8 hours ago - 1.62 thousand downloads last month - 26,030 stars on GitHub - 1 maintainer
jax-cuda12-plugin 0.4.28
JAX Plugin for NVIDIA GPUs
7 versions - Latest release: about 8 hours ago - 1.91 thousand downloads last month - 26,030 stars on GitHub - 4 maintainers
sgmcmcjax 0.2.13
SGMCMC samplers in JAX
18 versions - Latest release: 9 months ago - 1 dependent repositories - 178 downloads last month - 91 stars on GitHub - 2 maintainers
invrs-gym 0.10.3
A collection of inverse design challenges
26 versions - Latest release: 24 days ago - 480 downloads last month - 75 stars on GitHub - 1 maintainer
fmmax 0.8.0
Fourier modal method with Jax
15 versions - Latest release: about 1 month ago - 1 dependent package - 1 dependent repositories - 480 downloads last month - 75 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
42 versions - Latest release: 10 days ago - 147 dependent packages - 3,260 dependent repositories - 4.03 million downloads last month - 5,545 stars on GitHub - 7 maintainers
flowmc 0.3.3
Normalizing flow exhanced sampler in jax
22 versions - Latest release: 3 days ago - 3.82 thousand downloads last month - 173 stars on GitHub - 2 maintainers
coordinax 0.4.3
Vectors in JAX
8 versions - Latest release: about 1 month ago - 94 downloads last month - 3 stars on GitHub - 3 maintainers
in-transformers 1.0.0
State-of-the-art Machine Learning for JAX, PyTorch and TensorFlow
1 version - Latest release: over 1 year ago - 16 downloads last month - 125,725 stars on GitHub - 1 maintainer
grouped-query-attention-pytorch 0.2.0
grouped-query-attention-pytorch
3 versions - Latest release: 9 months ago - 200 downloads last month - 125,725 stars on GitHub - 2 maintainers
transformers-machinify 4.27.0.dev0
State-of-the-art Machine Learning for JAX, PyTorch and TensorFlow
1 version - Latest release: about 1 year ago - 1 dependent package - 47 downloads last month - 125,725 stars on GitHub - 2 maintainers
axlearn 0.0.1
AXLearn
3 versions - Latest release: 9 months ago - 118 downloads last month - 125,725 stars on GitHub - 2 maintainers
Top 0.1% on pypi.org
transformers 4.40.2
State-of-the-art Machine Learning for JAX, PyTorch and TensorFlow
141 versions - Latest release: 4 days ago - 1,523 dependent packages - 31,800 dependent repositories - 27.1 million downloads last month - 125,725 stars on GitHub - 6 maintainers
transformers-phobert 3.1.2
State-of-the-art Natural Language Processing for TensorFlow 2.0 and PyTorch. Note that the tokeni...
2 versions - Latest release: over 3 years ago - 1 dependent repositories - 21 downloads last month - 125,660 stars on GitHub - 1 maintainer
hf-argparser 0.1.1
Standalone version of the HuggingFace ArgParser from the Transformers library
2 versions - Latest release: 7 months ago - 1 dependent repositories - 19 downloads last month - 125,358 stars on GitHub - 2 maintainers
sofima 0.1
Scalable Optical Flow-based Image Montaging and Alignment
1 version - Latest release: about 2 years ago - 1 dependent repositories - 9 downloads last month - 47 stars on GitHub - 2 maintainers
jax-dimenet 1.0.0
DimeNet++ in Jax.
1 version - Latest release: about 2 years ago - 1 dependent repositories - 13 downloads last month - 15 stars on GitHub - 2 maintainers
piscis 0.1.2
An automatic deep learning algorithm for spot detection in fluorescence microscopy images.
6 versions - Latest release: about 15 hours ago - 1 dependent repositories - 55 downloads last month - 28,003 stars on GitHub - 2 maintainers
jax-chmc 0.0.7
An implementation of a family of MCMC methods on implicitly defined manifolds
5 versions - Latest release: 2 months ago - 40 downloads last month - 0 stars on GitHub - 2 maintainers
pcgym 0.1.6
Reinforcement learning suite of process control problems.
11 versions - Latest release: 23 days ago - 645 downloads last month - 16 stars on GitHub - 2 maintainers
roger 3.0.5
Runoff Generation Research - a process-based hydrological toolbox model in Python
7 versions - Latest release: about 1 month ago - 199 downloads last month - 2 stars on GitHub - 2 maintainers
pyhgf 0.0.17
The generalized, nodalized HGF for predictive coding.
17 versions - Latest release: 19 days ago - 223 downloads last month - 35 stars on GitHub - 2 maintainers
jax-fdm 0.8.4
Auto-differentiable and hardware-accelerated force density method
24 versions - Latest release: about 17 hours ago - 275 downloads last month - 77 stars on GitHub - 1 maintainer
Top 0.4% on pypi.org
diffusers 0.27.2
State-of-the-art diffusion in PyTorch and JAX.
74 versions - Latest release: about 2 months ago - 119 dependent packages - 5,162 dependent repositories - 2.41 million downloads last month - 20,155 stars on GitHub - 4 maintainers
ivy-mech 1.1.9
Mechanics functions with end-to-end support for machine learning developers, written in Ivy
11 versions - Latest release: over 2 years ago - 5 dependent packages - 4 dependent repositories - 170 downloads last month - 14 stars on GitHub - 2 maintainers
aws-fortuna 10.7.8
A Library for Uncertainty Quantification.
44 versions - Latest release: over 1 year ago - 2 dependent repositories - 602 downloads last month - 853 stars on GitHub - 3 maintainers
Top 2.3% on pypi.org
trax 1.4.1
Trax
24 versions - Latest release: over 2 years ago - 3 dependent packages - 64 dependent repositories - 3.96 thousand downloads last month - 7,963 stars on GitHub - 2 maintainers
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: 21 days ago - 28 dependent packages - 8,632 dependent repositories - 9.08 million downloads last month - 2,792 stars on GitHub - 3 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 - 2.07 thousand downloads last month - 2,771 stars on GitHub - 3 maintainers
penzai 0.1.1
Penzai: A JAX research toolkit for building, editing, and visualizing neural networks.
2 versions - Latest release: 6 days ago - 1.76 thousand downloads last month - 1,445 stars on GitHub - 4 maintainers
np-family 0.2.0
Neural Processes implementations in JAX and PyTorch
2 versions - Latest release: almost 2 years ago - 1 dependent repositories - 15 downloads last month - 3 stars on GitHub - 1 maintainer
autobnn 0.0.2.dev0
Package for training Gaussian process-like Bayesian Neural Networks with composite structure.
2 versions - Latest release: about 1 month ago - 250 downloads last month - 4,135 stars on GitHub - 2 maintainers
Top 8.1% on pypi.org
inference-gym 0.0.4
The Inference Gym is the place to exercise inference methods to help make them faster, leaner and...
5 versions - Latest release: over 3 years ago - 115 dependent repositories - 60 downloads last month - 4,135 stars on GitHub - 3 maintainers
Top 4.6% on pypi.org
mctx 0.0.5
Monte Carlo tree search in JAX.
4 versions - Latest release: 6 months ago - 1 dependent package - 4 dependent repositories - 1.71 thousand downloads last month - 2,211 stars on GitHub - 1 maintainer
jax-resnet 0.0.4
Framework-agnostic library for checking array shapes at runtime.
2 versions - Latest release: almost 3 years ago - 1 dependent package - 2 dependent repositories - 53 downloads last month - 101 stars on GitHub - 2 maintainers
flax-extra 0.0.2
The package provides extra flexibility to Flax using ideas originated at Trax
3 versions - Latest release: over 2 years ago - 1 dependent repositories - 33 downloads last month - 1 stars on GitHub - 2 maintainers
qarray 1.0.12
Qarray, a GPU accelerated quantum dot array simulator, leveraging parallelised Rust and JAX XLA a...
12 versions - Latest release: 7 days ago - 261 downloads last month - 4 stars on GitHub - 2 maintainers
cryojax 0.2.3
Cryo-EM image simulation and analysis powered by JAX
5 versions - Latest release: 29 days ago - 296 downloads last month - 26 stars on GitHub - 2 maintainers
Top 5.5% on pypi.org
merlin-dataloader 23.8.0
Merlin Dataloader
9 versions - Latest release: 8 months ago - 1 dependent package - 4 dependent repositories - 4.33 thousand downloads last month - 396 stars on GitHub - 2 maintainers
dejax 0.1.3
Accelerated replay buffers in JAX.
4 versions - Latest release: over 1 year ago - 24 downloads last month - 38 stars on GitHub - 2 maintainers
Top 9.5% on pypi.org
rax 0.3.0
Learning-to-Rank using JAX.
4 versions - Latest release: 12 months ago - 1 dependent repositories - 6.67 thousand downloads last month - 305 stars on GitHub - 2 maintainers
autobound 0.1.4
AutoBound automatically computes upper and lower bounds on functions.
5 versions - Latest release: 6 months ago - 61 downloads last month - 339 stars on GitHub - 1 maintainer
Top 2.7% on pypi.org
clu 0.0.12
Set of libraries for ML training loops in JAX.
14 versions - Latest release: 29 days ago - 9 dependent packages - 350 dependent repositories - 2.52 million downloads last month - 295 stars on GitHub - 4 maintainers
Top 2.4% on pypi.org
jaxopt 0.8.3
Hardware accelerated, batchable and differentiable optimizers in JAX.
17 versions - Latest release: 4 months ago - 35 dependent packages - 60 dependent repositories - 114 thousand downloads last month - 891 stars on GitHub - 1 maintainer
Top 4.7% on pypi.org
neural-tangents 0.6.5
Fast and Easy Infinite Neural Networks in Python
29 versions - Latest release: 5 months ago - 15 dependent repositories - 2.97 thousand downloads last month - 2,227 stars on GitHub - 2 maintainers
paxml 1.4.0
Framework to configure and run machine learning experiments on top of Jax.
12 versions - Latest release: about 1 month ago - 547 downloads last month - 397 stars on GitHub - 1 maintainer
jaxonnxruntime 0.3.0
Jaxonnxruntime: JAX based ONNX Runtime.
4 versions - Latest release: 9 months ago - 146 downloads last month - 71 stars on GitHub - 2 maintainers
Top 0.5% on pypi.org
etils 1.8.0
Collection of common python utils
27 versions - Latest release: about 2 months ago - 17 dependent packages - 567 dependent repositories - 3.62 million downloads last month - 170 stars on GitHub - 1 maintainer
Top 8.4% on pypi.org
dynamax 0.1.3
Dynamic State Space Models in JAX.
9 versions - Latest release: 2 months ago - 1 dependent package - 1 dependent repositories - 847 downloads last month - 617 stars on GitHub - 2 maintainers
Top 9.3% on pypi.org
jax-cfd 0.2.0
Computational Fluid Dynamics in JAX
2 versions - Latest release: almost 2 years ago - 1 dependent repositories - 2.14 thousand downloads last month - 644 stars on GitHub - 1 maintainer
somap 0.1.1
Self-Organizing Maps in python
3 versions - Latest release: 2 days ago - 121 downloads last month - 9 stars on GitHub - 2 maintainers
Top 2.6% on pypi.org
brax 0.10.3
A differentiable physics engine written in JAX.
27 versions - Latest release: about 2 months ago - 11 dependent packages - 24 dependent repositories - 9.57 thousand downloads last month - 2,074 stars on GitHub - 4 maintainers
levy-stable-jax 0.2.0
Implementation of the Lévy alpha-stable distributions for Jax.
5 versions - Latest release: 1 day ago - 462 downloads last month - 0 stars on GitHub - 2 maintainers
d-brax-jumpy 0.0.2 removed
Common backend for JAX or numpy.
2 versions - Latest release: about 2 years ago - 1,431 stars on GitHub
unxt 0.10.3
Quantities in JAX
14 versions - Latest release: 5 days ago - 1.12 thousand downloads last month - 4 stars on GitHub - 4 maintainers
sigkerax 0.2.1
Signature kernel computations via PDE solvers in JAX
13 versions - Latest release: 7 months ago - 131 downloads last month - 3 stars on GitHub - 2 maintainers
invrs-opt 0.5.1
Algorithms for inverse design
13 versions - Latest release: about 1 month ago - 2 dependent packages - 503 downloads last month - 2 stars on GitHub - 2 maintainers
jaxsim 0.2.0
A differentiable physics engine and multibody dynamics library for control and robot learning.
76 versions - Latest release: about 1 month ago - 815 downloads last month - 44 stars on GitHub - 4 maintainers
Top 1.8% on pypi.org
chex 0.1.86
Chex: Testing made fun, in JAX!
24 versions - Latest release: about 2 months ago - 61 dependent packages - 1,399 dependent repositories - 9.52 million downloads last month - 716 stars on GitHub - 3 maintainers
patato 0.5.14
A Python Toolkit for the analysis of photoacoustic tomography data.
24 versions - Latest release: 5 months ago - 1 dependent repositories - 3.53 thousand downloads last month - 9 stars on GitHub - 1 maintainer
coola 0.6.1
A library to check if two complex/nested objects are equal or not
50 versions - Latest release: 7 days ago - 9 dependent packages - 3 dependent repositories - 201 thousand downloads last month - 2 stars on GitHub - 1 maintainer
Top 1.7% on pypi.org
tensorly 0.8.1
Tensor learning in Python.
18 versions - Latest release: about 1 year ago - 23 dependent packages - 212 dependent repositories - 30.1 thousand downloads last month - 1,495 stars on GitHub - 1 maintainer
tensorly-musco 0.4.5
Tensor learning in Python.
1 version - Latest release: over 4 years ago - 1 dependent repositories - 18 downloads last month - 1,495 stars on GitHub - 2 maintainers
jaxnet 0.2.7
Neural Nets for JAX
12 versions - Latest release: over 3 years ago - 1 dependent repositories - 134 downloads last month - 183 stars on GitHub - 2 maintainers
fjformer 0.0.51
Embark on a journey of paralleled/unparalleled computational prowess with FJFormer - an arsenal o...
42 versions - Latest release: 4 days ago - 1 dependent package - 1.19 thousand downloads last month - 21 stars on GitHub - 2 maintainers
jax-fem 0.0.4
GPU accelerated Finite element analysis package in JAX.
4 versions - Latest release: 20 days ago - 193 downloads last month - 151 stars on GitHub - 1 maintainer
Top 9.9% on pypi.org
pgx 6.0.0
GPU/TPU-accelerated parallel game simulators for reinforcement learning (RL)
69 versions - Latest release: about 1 year ago - 1 dependent package - 1 dependent repositories - 2.2 thousand downloads last month - 319 stars on GitHub - 1 maintainer
Top 9.5% on pypi.org
squirrel-core 0.19.8
Squirrel is a Python library that enables ML teams to share, load, and transform data in a collab...
107 versions - Latest release: 4 months ago - 2 dependent repositories - 867 downloads last month - 279 stars on GitHub - 2 maintainers
quaxed 0.4.2
Array-API JAX compatibility
13 versions - Latest release: 2 days ago - 3.9 thousand downloads last month - 2 stars on GitHub - 2 maintainers
solstice-jax 0.0.1
Solstice, a library for creating and scaling experiments in JAX.
1 version - Latest release: almost 2 years ago - 19 downloads last month - 11 stars on GitHub - 2 maintainers
Top 1.5% on pypi.org
numpyro 0.14.0
Pyro PPL on NumPy
27 versions - Latest release: 2 months ago - 47 dependent packages - 123 dependent repositories - 143 thousand downloads last month - 2,049 stars on GitHub - 1 maintainer
Top 6.0% on pypi.org
gpjax 0.8.2 💰
Gaussian processes in JAX.
58 versions - Latest release: about 2 months ago - 3 dependent packages - 2 dependent repositories - 1.27 thousand downloads last month - 387 stars on GitHub - 2 maintainers
jimgw 0.1.1
Gravitatioanl wave data analysis tool in Jax
4 versions - Latest release: 8 months ago - 56 downloads last month - 49 stars on GitHub - 1 maintainer
Top 0.7% on pypi.org
wandb 0.17.0
A CLI and library for interacting with the Weights & Biases API.
266 versions - Latest release: 2 days ago - 401 dependent packages - 9,299 dependent repositories - 12.8 million downloads last month - 8,194 stars on GitHub - 8 maintainers
Top 4.9% on pypi.org
dm-pix 0.4.2
PIX is an image processing library in JAX, for JAX.
11 versions - Latest release: 5 months ago - 73 dependent repositories - 2.89 thousand downloads last month - 353 stars on GitHub - 6 maintainers
Top 9.2% on pypi.org
dawdreamer 0.8.0
An audio-processing Python library supporting core DAW features
28 versions - Latest release: 6 months ago - 1 dependent repositories - 1.41 thousand downloads last month - 821 stars on GitHub - 1 maintainer
agjax 0.3.1
A jax wrapper for autograd-differentiable functions.
3 versions - Latest release: 3 days ago - 1 dependent package - 1 dependent repositories - 388 downloads last month - 4 maintainers
hidiffusion 0.1.8
HiDiffusion: A training-free method to increase the resolution and speed of diffusion models.
6 versions - Latest release: 3 days ago - 2.07 thousand downloads last month - 22,366 stars on GitHub - 2 maintainers
Top 0.2% on pypi.org
jax 0.4.26
Differentiate, compile, and transform Numpy code.
160 versions - Latest release: about 1 month ago - 585 dependent packages - 6,336 dependent repositories - 6.32 million downloads last month - 28,003 stars on GitHub - 10 maintainers
Top 2.6% on pypi.org
distrax 0.1.5
Distrax: Probability distributions in JAX.
9 versions - Latest release: 6 months ago - 19 dependent packages - 214 dependent repositories - 4.88 million downloads last month - 509 stars on GitHub - 3 maintainers
triton-model-navigator 0.8.1
Triton Model Navigator: An inference toolkit for optimizing and deploying machine learning models...
23 versions - Latest release: about 1 month ago - 1 dependent repositories - 2.24 thousand downloads last month - 157 stars on GitHub - 2 maintainers
jax-am 0.0.3
GPU-accelerated simulation toolbox for additive manufacturing based on JAX.
3 versions - Latest release: 11 months ago - 32 downloads last month - 220 stars on GitHub - 1 maintainer
ramsey 0.2.1
Probabilistic deep learning using JAX
5 versions - Latest release: 7 months ago - 1 dependent repositories - 46 downloads last month - 10 stars on GitHub - 1 maintainer
easydel 0.0.64
An open-source library to make training faster and more optimized in Jax/Flax
44 versions - Latest release: 3 days ago - 758 downloads last month - 66 stars on GitHub - 2 maintainers
jax-tools 1.0.140
Jax common tools library
61 versions - Latest release: 3 days ago - 971 downloads last month - 2 maintainers
hf-image-uploader 0.0.2
State-of-the-art diffusion in PyTorch and JAX.
2 versions - Latest release: 8 months ago - 10 downloads last month - 10 stars on GitHub - 2 maintainers
Top 0.6% on pypi.org
pytorch-pretrained-bert 0.6.2
PyTorch version of Google AI BERT model with script to load Google pre-trained models
10 versions - Latest release: about 5 years ago - 9 dependent packages - 940 dependent repositories - 25.4 thousand downloads last month - 125,524 stars on GitHub - 1 maintainer