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
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
pyhf 0.7.6
pure-Python HistFactory implementation with tensors and autodiff47 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 compatibility5 versions - Latest release: 2 months ago - 47 downloads last month - 2 stars on GitHub - 2 maintainers
Top 4.3% on pypi.org
23 versions - Latest release: 2 months ago - 23 dependent packages - 65 dependent repositories - 220 thousand downloads last month - 115 stars on GitHub - 2 maintainers
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 Models15 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
11 versions - Latest release: over 2 years ago - 3 dependent packages - 4 dependent repositories - 167 downloads last month - 71 stars on GitHub - 2 maintainers
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 JaxGaussianProcesses418 versions - Latest release: about 7 hours ago - 4.33 thousand downloads last month - 2 maintainers
Top 0.2% on pypi.org
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
jaxlib 0.4.28
XLA library for JAX55 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 GPUs7 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 GPUs7 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 JAX18 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 challenges26 versions - Latest release: 24 days ago - 480 downloads last month - 75 stars on GitHub - 1 maintainer
fmmax 0.8.0
Fourier modal method with Jax15 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
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
flax 0.8.3
Flax: A neural network library for JAX designed for flexibility42 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 jax22 versions - Latest release: 3 days ago - 3.82 thousand downloads last month - 173 stars on GitHub - 2 maintainers
coordinax 0.4.3
Vectors in JAX8 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 TensorFlow1 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-pytorch3 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 TensorFlow1 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
AXLearn3 versions - Latest release: 9 months ago - 118 downloads last month - 125,725 stars on GitHub - 2 maintainers
Top 0.1% on pypi.org
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 4.40.2
State-of-the-art Machine Learning for JAX, PyTorch and TensorFlow141 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 library2 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 Alignment1 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 manifolds5 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 Python7 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 method24 versions - Latest release: about 17 hours ago - 275 downloads last month - 77 stars on GitHub - 1 maintainer
Top 0.4% on pypi.org
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
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 Ivy11 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
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
trax 1.4.1
Trax24 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
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
thinc 9.0.0
A refreshing functional take on deep learning, compatible with your favorite libraries232 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
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
thinc-gpu-ops 0.0.4
CUDA kernels for Thinc4 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 PyTorch2 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
5 versions - Latest release: over 3 years ago - 115 dependent repositories - 60 downloads last month - 4,135 stars on GitHub - 3 maintainers
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
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
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 Trax3 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 JAX5 versions - Latest release: 29 days ago - 296 downloads last month - 26 stars on GitHub - 2 maintainers
Top 5.5% on pypi.org
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
merlin-dataloader 23.8.0
Merlin Dataloader9 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
4 versions - Latest release: 12 months ago - 1 dependent repositories - 6.67 thousand downloads last month - 305 stars on GitHub - 2 maintainers
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
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
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
17 versions - Latest release: 4 months ago - 35 dependent packages - 60 dependent repositories - 114 thousand downloads last month - 891 stars on GitHub - 1 maintainer
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
29 versions - Latest release: 5 months ago - 15 dependent repositories - 2.97 thousand downloads last month - 2,227 stars on GitHub - 2 maintainers
neural-tangents 0.6.5
Fast and Easy Infinite Neural Networks in Python29 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
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
etils 1.8.0
Collection of common python utils27 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
9 versions - Latest release: 2 months ago - 1 dependent package - 1 dependent repositories - 847 downloads last month - 617 stars on GitHub - 2 maintainers
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
2 versions - Latest release: almost 2 years ago - 1 dependent repositories - 2.14 thousand downloads last month - 644 stars on GitHub - 1 maintainer
jax-cfd 0.2.0
Computational Fluid Dynamics in JAX2 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 python3 versions - Latest release: 2 days ago - 121 downloads last month - 9 stars on GitHub - 2 maintainers
Top 2.6% on pypi.org
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
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 JAX14 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 JAX13 versions - Latest release: 7 months ago - 131 downloads last month - 3 stars on GitHub - 2 maintainers
invrs-opt 0.5.1
Algorithms for inverse design13 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
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
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 not50 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
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 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 JAX12 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
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
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
107 versions - Latest release: 4 months ago - 2 dependent repositories - 867 downloads last month - 279 stars on GitHub - 2 maintainers
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 compatibility13 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
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
numpyro 0.14.0
Pyro PPL on NumPy27 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
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
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 Jax4 versions - Latest release: 8 months ago - 56 downloads last month - 49 stars on GitHub - 1 maintainer
Top 0.7% on pypi.org
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
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
11 versions - Latest release: 5 months ago - 73 dependent repositories - 2.89 thousand downloads last month - 353 stars on GitHub - 6 maintainers
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
28 versions - Latest release: 6 months ago - 1 dependent repositories - 1.41 thousand downloads last month - 821 stars on GitHub - 1 maintainer
dawdreamer 0.8.0
An audio-processing Python library supporting core DAW features28 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
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
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
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
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 JAX5 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/Flax44 versions - Latest release: 3 days ago - 758 downloads last month - 66 stars on GitHub - 2 maintainers
jax-tools 1.0.140
Jax common tools library61 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
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
pytorch-pretrained-bert 0.6.2
PyTorch version of Google AI BERT model with script to load Google pre-trained models10 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
Related Keywords
machine-learning
136
python
130
deep-learning
121
pytorch
106
tensorflow
90
flax
51
numpy
48
neural-networks
43
learning
39
reinforcement-learning
33
machine
25
machine learning
21
nlp
20
gpu
20
optimization
19
transformer
18
deep learning
17
bayesian
17
bayesian-inference
16
hacktoberfest
16
ai
14
keras
14
data-science
14
deep
14
simulation
13
natural-language-processing
13
gaussian-processes
13
language-model
13
equinox
13
pretrained-models
12
mlops
12
mxnet
12
automatic-differentiation
12
artificial intelligence
11
artificial-intelligence
11
seq2seq
11
transformers
11
autodiff
11
pytorch-transformers
10
physics
10
neural-network
10
probabilistic-programming
10
speech-recognition
10
nlp-library
10
autograd
10
bert
10
statistics
10
language-models
10
model-hub
10
NLP
9
BERT
9
tensor
9
llm
9
JAX
9
GPT-2
8
dataset
8
datasets
8
functional-programming
8
quantum
8
inference
7
reproducibility
7
data
7
research
7
torch
7
pypi
7
array
7
ml
7
cupy
7
linear-algebra
6
speech
6
differentiable
6
vision
6
robotics
6
mlir
6
differentiable-programming
6
optax
6
computer-vision
6
probability
6
model-versioning
6
Wav2Vec2
5
ViT
5
deep-neural-networks
5
hyperparameter-tuning
5
mcmc
5
numpyro
5
sampling
5
gpt
5
stable
5
astrophysics
5
astronomy
5
diffusion
5
ivy
5
dm-haiku
5
rl
5
quantum-computing
5
spherical-harmonics
5
parallel
5
Machine Learning
5
vulkan
5
spirv
5