Ecosyste.ms: Packages

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

Top 1.8% on pypi.org
Top 0.1% downloads on pypi.org
Top 0.2% dependent packages on pypi.org
Top 0.3% dependent repos on pypi.org
Top 6.7% forks on pypi.org
Top 0.7% docker downloads on pypi.org

pypi.org : chex

Chex: Testing made fun, in JAX!

Registry - Source - Documentation - JSON
purl: pkg:pypi/chex
Keywords: jax, testing, debugging, python, machine, learning
License: Apache-2.0
Latest release: 3 months ago
First release: almost 4 years ago
Dependent packages: 97
Dependent repositories: 1,399
Downloads: 8,444,901 last month
Stars: 723 on GitHub
Forks: 41 on GitHub
Docker dependents: 191
Docker downloads: 25,529,857
See more repository details: repos.ecosyste.ms
Last synced: 9 days ago

lacss 0.9.0
Tools for cell segmentation
40 versions - Latest release: 27 days ago - 1.07 thousand downloads last month - 18 stars on GitHub - 1 maintainer
easydel 0.0.65
An open-source library to make training faster and more optimized in Jax/Flax
46 versions - Latest release: 27 days ago - 399 downloads last month - 66 stars on GitHub - 1 maintainer
Top 7.2% on pypi.org
jaxns 2.5.0
Nested Sampling in JAX
43 versions - Latest release: 28 days ago - 7 dependent packages - 9 dependent repositories - 2.95 thousand downloads last month - 125 stars on GitHub - 1 maintainer
boax 0.1.2
Boax is a Bayesian Optimization library for JAX.
7 versions - Latest release: 28 days ago - 56 downloads last month - 1 stars on GitHub - 1 maintainer
efax 1.15.3
Exponential families for JAX
95 versions - Latest release: 28 days ago - 2 dependent repositories - 810 downloads last month - 48 stars on GitHub - 1 maintainer
tjax 0.32.1
Tools for JAX.
157 versions - Latest release: 28 days ago - 2 dependent packages - 3 dependent repositories - 2.69 thousand downloads last month - 41 stars on GitHub - 1 maintainer
fjformer 0.0.57
Embark on a journey of paralleled/unparalleled computational prowess with FJFormer - an arsenal o...
53 versions - Latest release: 29 days ago - 1 dependent package - 1.51 thousand downloads last month - 21 stars on GitHub - 1 maintainer
google-vizier-dev 0.1.16.dev20240514144958
Open Source Vizier: Distributed service framework for blackbox optimization and research.
28 versions - Latest release: 29 days ago - 880 downloads last month - 1,181 stars on GitHub - 1 maintainer
protes 0.3.7
Method PROTES (PRobabilistic Optimizer with TEnsor Sampling) for derivative-free optimization of ...
16 versions - Latest release: 29 days ago - 1 dependent package - 1 dependent repositories - 288 downloads last month - 23 stars on GitHub - 1 maintainer
lazyqml 1.0.0
LazyQML benchmarking utility to test quantum machine learning models.
14 versions - Latest release: 29 days ago - 982 downloads last month - 1 maintainer
xminigrid 0.8.0
JAX-accelerated meta-reinforcement learning environments inspired by XLand and MiniGrid
10 versions - Latest release: 30 days ago - 264 downloads last month - 534 stars on GitHub - 1 maintainer
biorobot 0.1.9
The Bio-inspired Robotics Testbed (BRT).
10 versions - Latest release: about 1 month ago - 151 downloads last month - 1 stars on GitHub - 1 maintainer
moojoco 1.0.8
A unified framework for implementing and interfacing with MuJoCo and MuJoCo-XLA simulation enviro...
10 versions - Latest release: about 1 month ago - 1 dependent package - 340 downloads last month - 1 stars on GitHub - 1 maintainer
diffilqrax 0.0.1
Differentiable iLQR algorithm for dynamical systems
1 version - Latest release: about 1 month ago - 57 downloads last month - 1 maintainer
juxtapose 0.0.32
juxtapose: Multi-Person Pose Tracking Inference SDK with RTMDet, YOLOv8, GDino, RTMPose (ONNX) & ...
32 versions - Latest release: about 1 month ago - 1 dependent repositories - 627 downloads last month - 24 stars on GitHub - 1 maintainer
momaland 0.1.0 💰
A standard API for Multi-Objective Multi-Agent Decision making and a diverse set of reference env...
2 versions - Latest release: about 1 month ago - 47 downloads last month - 22 stars on GitHub - 3 maintainers
craftax 1.4.1
An open-world environment for training RL agents
10 versions - Latest release: about 1 month ago - 1.58 thousand downloads last month - 143 stars on GitHub - 1 maintainer
differt2d 0.2.0 💰
2D Toolbox for Differentiable Ray Tracing
5 versions - Latest release: about 1 month ago - 195 downloads last month - 1 stars on GitHub - 1 maintainer
tracr-pypi 1.0.0
Compiler from RASP to transformer weights
1 version - Latest release: about 1 month ago - 1 dependent package - 284 downloads last month - 462 stars on GitHub - 1 maintainer
Top 9.6% on pypi.org
scenvi 0.3.1 💰
Integration of scRNA-seq and spatial transcriptomics data
22 versions - Latest release: about 2 months ago - 412 downloads last month - 4,927 stars on GitHub - 1 maintainer
rlberry 0.7.3
An easy-to-use reinforcement learning library for research and education
16 versions - Latest release: about 2 months ago - 1 dependent package - 3 dependent repositories - 124 downloads last month - 159 stars on GitHub - 4 maintainers
paltax 1.0.1
Strong lensing package using jax
2 versions - Latest release: about 2 months ago - 457 downloads last month - 11 stars on GitHub - 1 maintainer
evox 0.8.1
evox
17 versions - Latest release: about 2 months ago - 358 downloads last month - 120 stars on GitHub - 1 maintainer
Top 5.8% on pypi.org
gymnax 0.0.8
JAX-compatible version of Open AI's gym environments
8 versions - Latest release: about 2 months ago - 6 dependent packages - 2 dependent repositories - 6.11 thousand downloads last month - 527 stars on GitHub - 1 maintainer
jax-moseq 0.2.2
9 versions - Latest release: about 2 months ago - 269 downloads last month - 3 stars on GitHub - 1 maintainer
dgenerate 3.4.5 💰
Batch image generation and manipulation tool supporting Stable Diffusion and related techniques /...
20 versions - Latest release: about 2 months ago - 332 downloads last month - 16 stars on GitHub - 1 maintainer
reinforced-lib 1.1.1
Reinforcement learning library
9 versions - Latest release: about 2 months ago - 123 downloads last month - 61 stars on GitHub - 2 maintainers
Top 3.8% on pypi.org
ott-jax 0.4.6
Optimal Transport Tools in JAX.
41 versions - Latest release: 2 months ago - 9 dependent packages - 11 dependent repositories - 37.9 thousand downloads last month - 456 stars on GitHub - 1 maintainer
apax 0.4.0
Atomistic Learned Potential Package in JAX
2 versions - Latest release: 2 months ago - 74 downloads last month - 28,003 stars on GitHub - 1 maintainer
Top 8.8% on pypi.org
praxis 1.4.0
Functionalities such as a layers for building neural networks in Jax.
12 versions - Latest release: 2 months ago - 1 dependent package - 1 dependent repositories - 17.5 thousand downloads last month - 153 stars on GitHub - 1 maintainer
Top 4.8% on pypi.org
google-vizier 0.1.16
Open Source Vizier: Distributed service framework for blackbox optimization and research.
38 versions - Latest release: 2 months ago - 4 dependent packages - 2 dependent repositories - 20.8 thousand downloads last month - 1,171 stars on GitHub - 1 maintainer
Top 9.0% on pypi.org
kfac-jax 0.0.6
A Jax package for approximate curvature estimation and optimization using KFAC.
5 versions - Latest release: 2 months ago - 1 dependent package - 1 dependent repositories - 500 downloads last month - 202 stars on GitHub - 1 maintainer
jaxmarl 0.0.3
Multi-Agent Reinforcement Learning with JAX
3 versions - Latest release: 2 months ago - 1 dependent repositories - 3.77 thousand downloads last month - 349 stars on GitHub - 1 maintainer
synthax 0.2.2
SynthAX: A Fast Modular Synthesizer in JAX
5 versions - Latest release: 2 months ago - 57 downloads last month - 35 stars on GitHub - 1 maintainer
Top 4.9% on pypi.org
jumanji 1.0.1
A diverse suite of scalable reinforcement learning environments in JAX
14 versions - Latest release: 2 months ago - 2 dependent packages - 9 dependent repositories - 2.24 thousand downloads last month - 534 stars on GitHub - 1 maintainer
mlorax 0.9.5
MLoRAx is a minimalist library for low-rank adaptation designd to effortlessly enable parameter-e...
6 versions - Latest release: 3 months ago - 19 downloads last month - 10 stars on GitHub - 1 maintainer
Top 0.3% on pypi.org
optax 0.2.2
A gradient processing and optimisation library in JAX.
20 versions - Latest release: 3 months ago - 227 dependent packages - 2,454 dependent repositories - 8.29 million downloads last month - 1,505 stars on GitHub - 3 maintainers
Top 9.2% on pypi.org
tf2jax 0.3.6
TF2JAX: Convert TensorFlow to JAX
10 versions - Latest release: 3 months ago - 1 dependent package - 1 dependent repositories - 3.23 thousand downloads last month - 96 stars on GitHub - 1 maintainer
Top 6.0% on pypi.org
gpjax 0.8.2 💰
Gaussian processes in JAX.
58 versions - Latest release: 3 months ago - 6 dependent packages - 2 dependent repositories - 988 downloads last month - 387 stars on GitHub - 2 maintainers
hj-reachability 0.6.0
Hamilton-Jacobi reachability analysis in JAX.
4 versions - Latest release: 3 months ago - 1 dependent repositories - 185 downloads last month - 84 stars on GitHub - 1 maintainer
dm-meltingpot 2.2.2
A suite of test scenarios for multi-agent reinforcement learning.
12 versions - Latest release: 3 months ago - 4 dependent packages - 447 downloads last month - 545 stars on GitHub - 2 maintainers
Top 0.5% on pypi.org
etils 1.8.0
Collection of common python utils
29 versions - Latest release: 3 months ago - 22 dependent packages - 567 dependent repositories - 3.25 million downloads last month - 175 stars on GitHub - 1 maintainer
dataclass-array 1.5.2
Dataclasses that behave like numpy arrays (with indexing, slicing, vectorization).
12 versions - Latest release: 3 months ago - 1 dependent repositories - 33.5 thousand downloads last month - 44 stars on GitHub - 1 maintainer
differt 0.0.10 💰
Differentiable Ray Tracing Toolbox for Radio Propagation Simulations
15 versions - Latest release: 3 months ago - 1.12 thousand downloads last month - 1 stars on GitHub - 1 maintainer
flashbax 0.1.2
Flashbax is an experience replay library oriented around JAX. Tailored to integrate seamlessly wi...
4 versions - Latest release: 3 months ago - 1 dependent package - 1 dependent repositories - 4.17 thousand downloads last month - 149 stars on GitHub - 1 maintainer
navix 0.4.0
Accelerated gridworld navigation with JAX for deep reinforcement learning
25 versions - Latest release: 3 months ago - 3 dependent packages - 761 downloads last month - 27 stars on GitHub - 1 maintainer
Top 5.3% on pypi.org
evosax 0.1.6
JAX-Based Evolution Strategies
16 versions - Latest release: 3 months ago - 7 dependent packages - 3 dependent repositories - 9.55 thousand downloads last month - 451 stars on GitHub - 1 maintainer
Top 1.9% on pypi.org
scvi-tools 1.1.2
Deep probabilistic analysis of single-cell omics data.
77 versions - Latest release: 3 months ago - 43 dependent packages - 41 dependent repositories - 19.5 thousand downloads last month - 1,113 stars on GitHub - 3 maintainers
scib-metrics 0.5.1
Accelerated and Python-only scIB metrics
19 versions - Latest release: 4 months ago - 3 dependent packages - 1 dependent repositories - 2.68 thousand downloads last month - 40 stars on GitHub - 2 maintainers
moojoco-env 1.0.1 removed
A unified framework for implementing and interfacing with MuJoCo and MuJoCo-XLA simulation enviro...
2 versions - Latest release: 4 months ago - 1 maintainer
mapc-sim 0.1.5
IEEE 802.11 MAPC (C-SR) simulator
6 versions - Latest release: 4 months ago - 63 downloads last month - 0 stars on GitHub - 1 maintainer
qdax 0.3.0
A Python Library for Quality-Diversity and NeuroEvolution
6 versions - Latest release: 5 months ago - 1 dependent repositories - 112 downloads last month - 243 stars on GitHub - 3 maintainers
bojaxns 1.1.1
Bayesian Optimisation with JAXNS
8 versions - Latest release: 5 months ago - 1 dependent repositories - 69 downloads last month - 0 stars on GitHub - 1 maintainer
queuinx 0.0.1
Queuinx: A library for performance evaluation in Jax
2 versions - Latest release: 5 months ago - 21 downloads last month - 3 stars on GitHub - 1 maintainer
helx 1.1.5
Interoperate among reinforcement learning libraries with jax, pytorch, gym and dm_env
12 versions - Latest release: 5 months ago - 55 downloads last month - 10 stars on GitHub - 1 maintainer
moss-rl 0.0.5
A Python library for Reinforcement Learning.
6 versions - Latest release: 5 months ago - 24 downloads last month - 2 stars on GitHub - 1 maintainer
dopamax 0.2.0
Reinforcement learning in pure JAX.
2 versions - Latest release: 6 months ago - 15 downloads last month - 10 stars on GitHub - 1 maintainer
alphafold-colabfold 2.3.6
An implementation of the inference pipeline of AlphaFold v2.3.1. This is a completely new model t...
22 versions - Latest release: 6 months ago - 1 dependent package - 4 dependent repositories - 62.5 thousand downloads last month - 1 stars on GitHub - 4 maintainers
dilax 0.1.6
Differentiable (binned) likelihoods in JAX.
6 versions - Latest release: 6 months ago - 19 downloads last month - 17 stars on GitHub - 1 maintainer
bayes-jones 0.0.1
Bayesian inference of Jones matrices.
1 version - Latest release: 6 months ago - 16 downloads last month - 0 stars on GitHub - 1 maintainer
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: 6 months ago - 2 dependent packages - 73 dependent repositories - 1.8 thousand downloads last month - 353 stars on GitHub - 3 maintainers
mubelnet 0.0.8
Deep Bayesian unsupervised decoder networks. Use poisson or multinomial belief networks to cluste...
1 version - Latest release: 7 months ago - 12 downloads last month - 0 stars on GitLab.com - 1 maintainer
Top 4.6% on pypi.org
mctx 0.0.5
Monte Carlo tree search in JAX.
4 versions - Latest release: 7 months ago - 3 dependent packages - 4 dependent repositories - 2.14 thousand downloads last month - 2,214 stars on GitHub - 1 maintainer
Top 2.6% on pypi.org
distrax 0.1.5
Distrax: Probability distributions in JAX.
9 versions - Latest release: 7 months ago - 25 dependent packages - 214 dependent repositories - 2.22 million downloads last month - 512 stars on GitHub - 3 maintainers
ued 0.1.0
Efficient baselines for autocurricula in JAX
8 versions - Latest release: 7 months ago - 50 downloads last month - 144 stars on GitHub - 1 maintainer
minimax-lib 0.1.0
Efficient baselines for autocurricula in JAX
2 versions - Latest release: 7 months ago - 31 downloads last month - 144 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
umaia 0.2
Toolbox for the processing and analysis of MALDI-MSI data
2 versions - Latest release: 8 months ago - 24 downloads last month - 6 stars on GitHub - 1 maintainer
ramsey 0.2.1
Probabilistic deep learning using JAX
5 versions - Latest release: 8 months ago - 1 dependent repositories - 28 downloads last month - 10 stars on GitHub - 1 maintainer
probabilistic-reconciliation 0.1.0
Probabilistic reconciliation of time series forecasts
5 versions - Latest release: 8 months ago - 36 downloads last month - 4 stars on GitHub - 1 maintainer
absl-extra 0.1.3
A wrapper to run and monitor absl app.
99 versions - Latest release: 8 months ago - 503 downloads last month - 0 stars on GitHub - 1 maintainer
cr-sparse 0.4.0 💰
Accelerated sparse representations and compressive sensing
13 versions - Latest release: 8 months ago - 2 dependent repositories - 87 downloads last month - 80 stars on GitHub - 1 maintainer
cr-wavelets 0.4.0 💰
Wavelets with JAX
3 versions - Latest release: 8 months ago - 1 dependent package - 1 dependent repositories - 52 downloads last month - 6 stars on GitHub - 1 maintainer
juxai-s2 3.0.0
JUX is a jax-accelerated engine for Lux-2022.
5 versions - Latest release: 9 months ago - 1 dependent package - 1 dependent repositories - 34 downloads last month - 18 stars on GitHub - 1 maintainer
pyrovelocity 0.1.3
A multivariate RNA Velocity model to estimate future cell states with uncertainty using probabili...
17 versions - Latest release: 9 months ago - 83 downloads last month - 35 stars on GitHub - 3 maintainers
tsuite 1.0
tsuite: Get your RL agent fixed today!
1 version - Latest release: 10 months ago - 7 downloads last month - 3 stars on GitHub - 1 maintainer
axlearn 0.0.1
AXLearn
3 versions - Latest release: 10 months ago - 140 downloads last month - 126,388 stars on GitHub - 1 maintainer
relaxed 0.4.0
Differentiable versions of common HEP operations.
15 versions - Latest release: 10 months ago - 5 dependent repositories - 125 downloads last month - 22 stars on GitHub - 1 maintainer
visu3d 1.5.3
3d geometry made easy.
10 versions - Latest release: 11 months ago - 2 dependent packages - 1 dependent repositories - 27.9 thousand downloads last month - 143 stars on GitHub - 1 maintainer
snaxlib 0.0.34
A simple deep learning library for JAX.
1 version - Latest release: 11 months ago - 10 downloads last month - 11 stars on GitHub - 1 maintainer
Top 2.2% on pypi.org
rlax 0.1.6
A library of reinforcement learning building blocks in JAX.
9 versions - Latest release: 12 months ago - 9 dependent packages - 190 dependent repositories - 1.88 million downloads last month - 1,193 stars on GitHub - 4 maintainers
haliax 1.0.1
Named Tensors for Legible Deep Learning in JAX
19 versions - Latest release: 12 months ago - 1 dependent package - 1 dependent repositories - 17.2 thousand downloads last month - 122 stars on GitHub - 1 maintainer
modularbayes 0.1.4
Modular Bayesian Inference.
3 versions - Latest release: 12 months ago - 14 downloads last month - 1 stars on GitHub - 1 maintainer
chemtrain 0.0.1
Training molecular dynamics potentials.
1 version - Latest release: 12 months ago - 23 downloads last month - 14 stars on GitHub - 1 maintainer
stamox 0.1.5
Accelerate Your Statistical Analysis with JAX.
12 versions - Latest release: about 1 year ago - 49 downloads last month - 9 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 - 4 dependent packages - 1 dependent repositories - 2.99 thousand downloads last month - 326 stars on GitHub - 1 maintainer
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
colabdesign 1.1.1
Making Protein Design accessible to all via Google Colab!
1 version - Latest release: about 1 year ago - 43 downloads last month - 1 maintainer
chemise 0.0.1a1
Wrapper for training flax models
2 versions - Latest release: about 1 year ago - 13 downloads last month - 1 stars on GitHub - 1 maintainer
precondition-opt 0.3.0
Preconditioning optimizers.
3 versions - Latest release: about 1 year ago - 1 dependent package - 34.3 thousand downloads last month - 25 stars on GitHub - 1 maintainer
Top 8.6% on pypi.org
alpa 0.2.3
Alpa automatically parallelizes large tensor computation graphs and runs them on a distributed cl...
13 versions - Latest release: over 1 year ago - 1 dependent repositories - 166 downloads last month - 2,989 stars on GitHub - 1 maintainer
coax 0.1.13
Plug-n-play reinforcement learning with Gymnasium and JAX
14 versions - Latest release: over 1 year ago - 1 dependent package - 1 dependent repositories - 164 downloads last month - 159 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
juxai2022 1.0
JUX is a jax-accelerated engine for Lux-2022.
1 version - Latest release: over 1 year ago - 7 downloads last month - 18 stars on GitHub - 1 maintainer
mogpjax 0.0.2 💰
A JaxLinOp library.
2 versions - Latest release: over 1 year ago - 1 dependent repositories - 9 downloads last month - 11 stars on GitHub - 2 maintainers
jestimator 0.3.3
Implementation of the Amos optimizer from the JEstimator lib.
5 versions - Latest release: over 1 year ago - 4 dependent repositories - 50 downloads last month - 77 stars on GitHub - 1 maintainer
helax 0.1.4
Python package for computing helicity amplitudes
4 versions - Latest release: over 1 year ago - 16 downloads last month - 0 stars on GitHub - 1 maintainer