Ecosyste.ms: Packages

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

Top 0.3% on pypi.org
Top 0.4% downloads on pypi.org
Top 0.1% dependent packages on pypi.org
Top 0.2% dependent repos on pypi.org
Top 0.7% docker downloads on pypi.org

pypi.org : optax

A gradient processing and optimisation library in JAX.

Registry - Source - Documentation - JSON
purl: pkg:pypi/optax
Keywords: python, machine learning, reinforcement-learning, machine-learning, optimization
License: Apache-2.0
Latest release: about 2 months ago
First release: almost 4 years ago
Dependent packages: 227
Dependent repositories: 2,454
Downloads: 8,582,798 last month
Stars: 1,503 on GitHub
Forks: 154 on GitHub
Docker dependents: 169
Docker downloads: 25,524,678
See more repository details: repos.ecosyste.ms
Last synced: 3 days ago

flowmc 0.3.4
Normalizing flow exhanced sampler in jax
23 versions - Latest release: 6 days ago - 1 dependent package - 3.54 thousand downloads last month - 174 stars on GitHub - 1 maintainer
f3dasm-optimize 1.5.2
f3dasm_optimize: Your one line description of the package
8 versions - Latest release: 6 days ago - 139 downloads last month - 0 stars on GitHub - 1 maintainer
prophetverse 0.1.1
9 versions - Latest release: 6 days ago - 836 downloads last month - 1 maintainer
jaxparrow 0.2.0
Computes the inversion of the cyclogeostrophic balance based on a variational formulation approac...
10 versions - Latest release: 6 days ago - 92 downloads last month - 4 stars on GitHub - 1 maintainer
nadl 1.3.0
Nasy's Deep Learning Toolkit
20 versions - Latest release: 7 days ago - 783 downloads last month - 1 maintainer
flowjax 12.2.0
Easy to use distributions, bijections and normalizing flows in JAX.
36 versions - Latest release: 7 days ago - 1 dependent package - 1 dependent repositories - 837 downloads last month - 68 stars on GitHub - 1 maintainer
easydel 0.0.65
An open-source library to make training faster and more optimized in Jax/Flax
45 versions - Latest release: 7 days ago - 797 downloads last month - 66 stars on GitHub - 1 maintainer
imt-ring 1.3.5
RING: Recurrent Inertial Graph-based Estimator
9 versions - Latest release: 8 days ago - 1 dependent package - 698 downloads last month - 0 stars on GitHub - 1 maintainer
jinns 0.8.7
Physics Informed Neural Network with JAX
28 versions - Latest release: 8 days ago - 392 downloads last month - 0 stars on GitLab.com - 2 maintainers
Top 7.2% on pypi.org
jaxns 2.5.0
Nested Sampling in JAX
42 versions - Latest release: 8 days ago - 7 dependent packages - 9 dependent repositories - 3.46 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: 8 days ago - 56 downloads last month - 1 stars on GitHub - 1 maintainer
tjax 0.32.1
Tools for JAX.
156 versions - Latest release: 9 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...
46 versions - Latest release: 9 days ago - 1 dependent package - 933 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.
23 versions - Latest release: 9 days ago - 1.15 thousand downloads last month - 1,179 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: 9 days ago - 1 dependent package - 1 dependent repositories - 289 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: 9 days ago - 982 downloads last month - 1 maintainer
Top 1.3% on pypi.org
tfx 1.15.1
TensorFlow Extended (TFX) is a TensorFlow-based general-purpose machine learning platform impleme...
98 versions - Latest release: 10 days ago - 4 dependent packages - 641 dependent repositories - 72.5 thousand downloads last month - 2,079 stars on GitHub - 3 maintainers
Top 1.5% on pypi.org
numpyro 0.15.0
Pyro PPL on NumPy
28 versions - Latest release: 10 days ago - 76 dependent packages - 123 dependent repositories - 145 thousand downloads last month - 2,057 stars on GitHub - 1 maintainer
Top 6.4% on pypi.org
netket 3.12.0
Netket : Machine Learning toolbox for many-body quantum systems.
62 versions - Latest release: 10 days ago - 5 dependent repositories - 1.59 thousand downloads last month - 498 stars on GitHub - 2 maintainers
xminigrid 0.8.0
JAX-accelerated meta-reinforcement learning environments inspired by XLand and MiniGrid
10 versions - Latest release: 10 days ago - 264 downloads last month - 534 stars on GitHub - 1 maintainer
Top 2.6% on pypi.org
brax 0.10.4
A differentiable physics engine written in JAX.
28 versions - Latest release: 10 days ago - 17 dependent packages - 24 dependent repositories - 10.6 thousand downloads last month - 2,082 stars on GitHub - 3 maintainers
nanodl 1.2.4.dev1 💰
A Jax-based library for designing and training transformer models from scratch.
6 versions - Latest release: 11 days ago - 110 downloads last month - 256 stars on GitHub - 1 maintainer
zenflow 0.1.2
Learn distributions with normalizing flows.
1 version - Latest release: 13 days ago - 0 stars on GitHub - 1 maintainer
mmpdenet 0.1.0
1 version - Latest release: 14 days ago - 1 maintainer
pycollimator 2.0.4
Collimator.ai core simulation engine and API client
72 versions - Latest release: 14 days ago - 521 downloads last month - 4 maintainers
lightcurver 1.0.6
A thorough structure for precise photometry and deconvolution of time series of wide field images.
7 versions - Latest release: 15 days ago - 178 downloads last month - 1 stars on GitHub - 1 maintainer
Top 2.1% on pypi.org
blackjax 1.2.1
Flexible and fast sampling in Python
24 versions - Latest release: 15 days ago - 15 dependent packages - 13 dependent repositories - 24.4 thousand downloads last month - 727 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
143 versions - Latest release: 17 days ago - 2,589 dependent packages - 31,800 dependent repositories - 28.1 million downloads last month - 126,450 stars on GitHub - 4 maintainers
juxtapose 0.0.32
juxtapose: Multi-Person Pose Tracking Inference SDK with RTMDet, YOLOv8, GDino, RTMPose (ONNX) & ...
32 versions - Latest release: 19 days ago - 1 dependent repositories - 627 downloads last month - 24 stars on GitHub - 1 maintainer
penzai 0.1.1
Penzai: A JAX research toolkit for building, editing, and visualizing neural networks.
2 versions - Latest release: 20 days ago - 1 dependent package - 1.76 thousand downloads last month - 1,445 stars on GitHub - 2 maintainers
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: 20 days ago - 22 downloads last month - 22 stars on GitHub - 3 maintainers
nndp 0.1.0
Dynamic Programming using Neural Networks
9 versions - Latest release: 20 days ago - 183 downloads last month - 2,806 stars on GitHub - 1 maintainer
qadence 1.5.2
Pasqal interface for circuit-based quantum computing SDKs
25 versions - Latest release: 20 days ago - 1.66 thousand downloads last month - 55 stars on GitHub - 4 maintainers
meent 0.9.10
Electromagnetic simulation (RCWA) & optimization package in Python
47 versions - Latest release: 22 days ago - 1 dependent repositories - 367 downloads last month - 40 stars on GitHub - 1 maintainer
Top 3.2% on pypi.org
fedml 0.8.30 💰
A research and production integrated edge-cloud library for federated/distributed machine learnin...
350 versions - Latest release: 23 days ago - 1 dependent package - 9 dependent repositories - 30.9 thousand downloads last month - 3,930 stars on GitHub - 2 maintainers
craftax 1.4.1
An open-world environment for training RL agents
8 versions - Latest release: 23 days ago - 2.09 thousand downloads last month - 130 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: 23 days ago - 198 dependent packages - 3,260 dependent repositories - 4.22 million downloads last month - 5,583 stars on GitHub - 5 maintainers
differt2d 0.2.0 💰
2D Toolbox for Differentiable Ray Tracing
2 versions - Latest release: 24 days ago - 72 downloads last month - 1 stars on GitHub - 1 maintainer
evermore 0.2.6
Differentiable (binned) likelihoods in JAX.
7 versions - Latest release: 24 days ago - 115 downloads last month - 17 stars on GitHub - 1 maintainer
jax-relax 0.2.7
JAX-based Recourse Explanation Library
16 versions - Latest release: 26 days ago - 208 downloads last month - 6 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: 29 days 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: 29 days ago - 1 dependent package - 3 dependent repositories - 1.05 thousand downloads last month - 157 stars on GitHub - 4 maintainers
paltax 1.0.1
Strong lensing package using jax
2 versions - Latest release: 30 days ago - 457 downloads last month - 11 stars on GitHub - 1 maintainer
evox 0.8.1
evox
17 versions - Latest release: about 1 month ago - 343 downloads last month - 120 stars on GitHub - 1 maintainer
stadion 1.0.1
Causal Modeling with Stationary Diffusions
2 versions - Latest release: about 1 month ago - 112 downloads last month - 3 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 - 213 downloads last month - 40 stars on GitHub - 1 maintainer
jax-jynx 0.1.3
A neural network library using jax
4 versions - Latest release: about 1 month ago - 44 downloads last month - 1 maintainer
coix 0.1.0
Inference Combinators in JAX
2 versions - Latest release: about 1 month ago - 104 downloads last month - 38 stars on GitHub - 1 maintainer
horqrux 0.6.2
Jax-based quantum state vector simulator.
6 versions - Latest release: about 1 month ago - 1 dependent package - 1.1 thousand downloads last month - 1 maintainer
dgenerate 3.4.5 💰
Batch image generation and manipulation tool supporting Stable Diffusion and related techniques /...
18 versions - Latest release: about 1 month ago - 152 downloads last month - 16 stars on GitHub - 1 maintainer
invrs-gym 0.10.3
A collection of inverse design challenges
26 versions - Latest release: about 1 month ago - 1 dependent package - 525 downloads last month - 75 stars on GitHub - 1 maintainer
phlash 1.0.2
Bayesian inference of population size history from recombining sequence data.
3 versions - Latest release: about 1 month ago - 28 downloads last month - 7 stars on GitHub - 1 maintainer
trajatlas 0.1.4
TrajAtlas, a trajectory-centric framework reveals differentiation heterogenity among cells, genes...
2 versions - Latest release: about 1 month ago - 76 downloads last month - 0 stars on GitHub - 1 maintainer
clax 0.0.1
Prebuilt jax classifiers
1 version - Latest release: about 1 month ago - 151 downloads last month - 1 maintainer
reinforced-lib 1.1.1
Reinforcement learning library
9 versions - Latest release: about 1 month 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: about 1 month ago - 9 dependent packages - 11 dependent repositories - 20 thousand downloads last month - 456 stars on GitHub - 1 maintainer
sbx-rl 0.15.0
Jax version of Stable Baselines, implementations of reinforcement learning algorithms.
19 versions - Latest release: about 1 month ago - 1 dependent repositories - 736 downloads last month - 257 stars on GitHub - 1 maintainer
jo3mnist 0.0.5
6 versions - Latest release: about 1 month ago - 110 downloads last month - 1 maintainer
apax 0.4.0
Atomistic Learned Potential Package in JAX
2 versions - Latest release: about 1 month ago - 460 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: about 1 month ago - 1 dependent package - 1 dependent repositories - 6.6 thousand downloads last month - 146 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: about 1 month ago - 4 dependent packages - 2 dependent repositories - 19.8 thousand downloads last month - 1,171 stars on GitHub - 1 maintainer
bayeux-ml 0.1.11
Stitching together probabilistic models and inference.
12 versions - Latest release: about 2 months ago - 4.82 thousand downloads last month - 99 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: about 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: about 2 months ago - 1 dependent repositories - 4.9 thousand downloads last month - 342 stars on GitHub - 1 maintainer
Top 1.7% on pypi.org
deepchem 2.8.0
Deep learning models for drug discovery, quantum chemistry, and the life sciences.
861 versions - Latest release: about 2 months ago - 8 dependent packages - 55 dependent repositories - 27 thousand downloads last month - 5,164 stars on GitHub - 4 maintainers
Top 0.3% on pypi.org
optuna 3.6.1 💰
A hyperparameter optimization framework
62 versions - Latest release: about 2 months ago - 386 dependent packages - 2,141 dependent repositories - 2.61 million downloads last month - 9,797 stars on GitHub - 7 maintainers
gpfy 0.0.2
Gaussian process with spherical harmonic features in JAX
2 versions - Latest release: about 2 months ago - 31 downloads last month - 11 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: about 2 months ago - 2 dependent packages - 9 dependent repositories - 2.72 thousand downloads last month - 534 stars on GitHub - 1 maintainer
kooplearn 1.1.3
A package to learn Koopman operators
49 versions - Latest release: about 2 months ago - 2 dependent repositories - 445 downloads last month - 33 stars on GitHub - 1 maintainer
last-asr 0.0.4
The LAttice-based Speech Transducer (LAST) library
4 versions - Latest release: about 2 months ago - 170 downloads last month - 37 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 - 1.1 thousand downloads last month - 387 stars on GitHub - 2 maintainers
femr 0.2.3
Framework for Electronic Medical Records. A python package for building models using EHR data.
23 versions - Latest release: 2 months ago - 1.21 thousand downloads last month - 81 stars on GitHub - 2 maintainers
slax 0.0.1
An SNN/RNN training package in JAX
1 version - Latest release: 2 months ago - 16 downloads last month - 1 stars on GitHub - 1 maintainer
crispio 0.0.2
Designing CRISPR-Cas guide RNAs in bacteria.
4 versions - Latest release: 2 months ago - 60 downloads last month - 0 stars on GitHub - 1 maintainer
nm-transformers 1.7.0.43401
State-of-the-art Machine Learning for JAX, PyTorch and TensorFlow
4 versions - Latest release: 2 months ago - 3 dependent packages - 1 dependent repositories - 177 downloads last month - 9 stars on GitHub - 1 maintainer
bayesmbar 0.0.4
Bayesian Multistate Bennett Acceptance Ratio Methods
3 versions - Latest release: 2 months ago - 23 downloads last month - 7 stars on GitHub - 1 maintainer
feedbax 0.1.2
Optimal feedback control + interventions in JAX.
2 versions - Latest release: 2 months ago - 13 downloads last month - 1 stars on GitHub - 1 maintainer
differt 0.0.10 💰
Differentiable Ray Tracing Toolbox for Radio Propagation Simulations
9 versions - Latest release: 2 months ago - 1.32 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: 2 months ago - 1 dependent package - 1 dependent repositories - 4.17 thousand downloads last month - 149 stars on GitHub - 1 maintainer
avici 1.0.5
Amortized Inference for Causal Structure Learning
6 versions - Latest release: 2 months ago - 1 dependent repositories - 62 downloads last month - 43 stars on GitHub - 1 maintainer
kira_llm 0.2.0
That's right, I'm Kira ✍️
5 versions - Latest release: 3 months ago - 45 downloads last month - 17 stars on GitHub - 1 maintainer
bayesnf 0.1.3
Scalable spatiotemporal prediction with Bayesian neural fields
5 versions - Latest release: 3 months ago - 165 downloads last month - 19 stars on GitHub - 1 maintainer
nm-transformers-nightly 1.7.0.20240304
State-of-the-art Machine Learning for JAX, PyTorch and TensorFlow
37 versions - Latest release: 3 months ago - 3 dependent packages - 228 downloads last month - 9 stars on GitHub - 1 maintainer
benchmark-mi 0.1.2
Estimators of mutual information and distributions used to benchmark them.
3 versions - Latest release: 3 months ago - 47 downloads last month - 24 stars on GitHub - 2 maintainers
Top 7.2% on pypi.org
totypes 0.6.4
Custom datatypes useful in a topology optimization context
13 versions - Latest release: 3 months ago - 3 dependent packages - 2 dependent repositories - 823 downloads last month - 1 maintainer
supergrad 0.1.0 removed
SuperGrad differentiable Hamiltonian simulator
1 version - Latest release: 3 months ago - 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 - 25.4 thousand downloads last month - 1,113 stars on GitHub - 3 maintainers
sbijax 0.2.0
Simulation-based inference in JAX
5 versions - Latest release: 3 months ago - 29 downloads last month - 10 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: 3 months ago - 1 dependent package - 1 dependent repositories - 847 downloads last month - 617 stars on GitHub - 2 maintainers
lanfactory 0.4.6
Package with convenience functions to train LANs
6 versions - Latest release: 3 months ago - 25 downloads last month - 11 stars on GitHub - 1 maintainer
Top 8.5% on pypi.org
pzflow 3.1.3
Probabilistic modeling of tabular data with normalizing flows.
29 versions - Latest release: 3 months ago - 2 dependent packages - 2 dependent repositories - 1.89 thousand downloads last month - 53 stars on GitHub - 1 maintainer
matsim-tools 0.0.18
MATSim Agent-Based Transportation Simulation Framework - official python tools
18 versions - Latest release: 3 months ago - 2 dependent repositories - 297 downloads last month - 46 stars on GitHub - 2 maintainers
spyx 0.1.19
Spyx: SNNs in JAX
62 versions - Latest release: 3 months ago - 1 dependent repositories - 248 downloads last month - 71 stars on GitHub - 1 maintainer
vqepy 1.0.1
An open source Python package for molecular and condensed matter VQE research
3 versions - Latest release: 3 months ago - 16 downloads last month - 1 stars on GitHub - 1 maintainer
dgmrf 0.0.1
Deep Gaussian Markov Random Fields and their extensions
1 version - Latest release: 3 months ago - 9 downloads last month - 0 stars on GitLab.com - 1 maintainer
surjectors 0.3.1
Surjection layers for density estimation with normalizing flows
4 versions - Latest release: 3 months ago - 1 dependent package - 1 dependent repositories - 84 downloads last month - 0 stars on GitHub - 1 maintainer
gtech-optimus 0.0.7
Optimus library for real-time marketing personalization using RL.
6 versions - Latest release: 3 months ago - 74 downloads last month - 12 stars on GitHub - 1 maintainer
graphiq 0.1.0
GraphiQ is a Python library for the simulation, design, and optimization of quantum photonic circ...
6 versions - Latest release: 3 months ago - 29 downloads last month - 10 stars on GitHub - 2 maintainers
fouriax 0.1.4
A jax port of auraloss
5 versions - Latest release: 3 months ago - 44 downloads last month - 3 stars on GitHub - 1 maintainer
qax 0.3.1
A JAX transform for writing things which pretend to be tensors
5 versions - Latest release: 3 months ago - 1 dependent package - 1 dependent repositories - 591 downloads last month - 41 stars on GitHub - 1 maintainer