Ecosyste.ms: Packages
An open API service providing package, version and dependency metadata of many open source software ecosystems and registries.
Top 1.4% on pypi.org
Top 0.6% downloads on pypi.org
Top 0.3% dependent packages on pypi.org
Top 0.4% dependent repos on pypi.org
Top 3.5% forks on pypi.org
Top 2.0% docker downloads on pypi.org
Top 0.6% downloads on pypi.org
Top 0.3% dependent packages on pypi.org
Top 0.4% dependent repos on pypi.org
Top 3.5% forks on pypi.org
Top 2.0% docker downloads on pypi.org
pypi.org : dm-haiku
Haiku is a library for building neural networks in JAX.
Registry
-
Source
- Documentation
- JSON
purl: pkg:pypi/dm-haiku
Keywords: deep-learning, deep-neural-networks, jax, machine-learning, neural-networks
License: Apache-2.0
Latest release: 3 months ago
First release: over 4 years ago
Dependent packages: 70
Dependent repositories: 949
Downloads: 205,857 last month
Stars: 2,690 on GitHub
Forks: 222 on GitHub
Docker dependents: 35
Docker downloads: 5,931
See more repository details: repos.ecosyste.ms
Last synced: 3 days ago
imt-ring 1.3.5
RING: Recurrent Inertial Graph-based Estimator9 versions - Latest release: 8 days ago - 1 dependent package - 698 downloads last month - 0 stars on GitHub - 1 maintainer
Top 7.2% on pypi.org
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
jaxns 2.5.0
Nested Sampling in JAX42 versions - Latest release: 8 days ago - 7 dependent packages - 9 dependent repositories - 3.46 thousand downloads last month - 125 stars on GitHub - 1 maintainer
Top 1.5% on pypi.org
28 versions - Latest release: 9 days ago - 76 dependent packages - 123 dependent repositories - 145 thousand downloads last month - 2,057 stars on GitHub - 1 maintainer
numpyro 0.15.0
Pyro PPL on NumPy28 versions - Latest release: 9 days ago - 76 dependent packages - 123 dependent repositories - 145 thousand downloads last month - 2,057 stars on GitHub - 1 maintainer
Top 9.3% on pypi.org
3 versions - Latest release: 10 days ago - 1 dependent repositories - 2.02 thousand downloads last month - 660 stars on GitHub - 1 maintainer
jax-cfd 0.2.1
Computational Fluid Dynamics in JAX3 versions - Latest release: 10 days ago - 1 dependent repositories - 2.02 thousand downloads last month - 660 stars on GitHub - 1 maintainer
declearn 2.5.0
Declearn - a python package for private decentralized learning.15 versions - Latest release: 10 days ago - 165 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: 18 days ago - 1 dependent repositories - 627 downloads last month - 24 stars on GitHub - 1 maintainer
nndp 0.1.0
Dynamic Programming using Neural Networks9 versions - Latest release: 20 days ago - 183 downloads last month - 2,806 stars on GitHub - 1 maintainer
Top 3.2% on pypi.org
350 versions - Latest release: 22 days ago - 1 dependent package - 9 dependent repositories - 30.9 thousand downloads last month - 3,930 stars on GitHub - 2 maintainers
fedml 0.8.30 💰
A research and production integrated edge-cloud library for federated/distributed machine learnin...350 versions - Latest release: 22 days ago - 1 dependent package - 9 dependent repositories - 30.9 thousand downloads last month - 3,930 stars on GitHub - 2 maintainers
tracr-pypi 1.0.0
Compiler from RASP to transformer weights1 version - Latest release: 24 days ago - 1 dependent package - 462 stars on GitHub - 1 maintainer
jax-relax 0.2.7
JAX-based Recourse Explanation Library16 versions - Latest release: 25 days ago - 208 downloads last month - 6 stars on GitHub - 1 maintainer
rlberry 0.7.3
An easy-to-use reinforcement learning library for research and education16 versions - Latest release: 29 days ago - 1 dependent package - 3 dependent repositories - 1.05 thousand downloads last month - 157 stars on GitHub - 4 maintainers
jaxspec 0.0.6
jaxspec is a bayesian spectral fitting library for X-ray astronomy.5 versions - Latest release: about 1 month ago - 143 downloads last month - 1 maintainer
reinforced-lib 1.1.1
Reinforcement learning library9 versions - Latest release: about 1 month ago - 123 downloads last month - 61 stars on GitHub - 2 maintainers
Top 9.0% on pypi.org
5 versions - Latest release: about 2 months ago - 1 dependent package - 1 dependent repositories - 500 downloads last month - 202 stars on GitHub - 1 maintainer
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
Top 2.6% on pypi.org
276 versions - Latest release: about 2 months ago - 56 dependent repositories - 8.61 thousand downloads last month - 9,071 stars on GitHub - 4 maintainers
syft 0.8.6 💰
Perform numpy-like analysis on data that remains in someone elses server276 versions - Latest release: about 2 months ago - 56 dependent repositories - 8.61 thousand downloads last month - 9,071 stars on GitHub - 4 maintainers
Top 1.7% on pypi.org
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
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
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
optuna 3.6.1 💰
A hyperparameter optimization framework62 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
Top 4.9% on pypi.org
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
jumanji 1.0.1
A diverse suite of scalable reinforcement learning environments in JAX14 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
Top 0.3% on pypi.org
20 versions - Latest release: about 2 months ago - 227 dependent packages - 2,454 dependent repositories - 8.58 million downloads last month - 1,503 stars on GitHub - 3 maintainers
optax 0.2.2
A gradient processing and optimisation library in JAX.20 versions - Latest release: about 2 months ago - 227 dependent packages - 2,454 dependent repositories - 8.58 million downloads last month - 1,503 stars on GitHub - 3 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
gpax 0.1.8
Gaussian processes in NumPyro and JAX12 versions - Latest release: 2 months ago - 1 dependent repositories - 317 downloads last month - 185 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 Learning6 versions - Latest release: 2 months ago - 1 dependent repositories - 62 downloads last month - 43 stars on GitHub - 1 maintainer
supergrad 0.1.0 removed
SuperGrad differentiable Hamiltonian simulator1 version - Latest release: 3 months ago - 1 maintainer
sbijax 0.2.0
Simulation-based inference in JAX5 versions - Latest release: 3 months ago - 29 downloads last month - 10 stars on GitHub - 1 maintainer
spyx 0.1.19
Spyx: SNNs in JAX62 versions - Latest release: 3 months ago - 1 dependent repositories - 248 downloads last month - 71 stars on GitHub - 1 maintainer
surjectors 0.3.1
Surjection layers for density estimation with normalizing flows4 versions - Latest release: 3 months ago - 1 dependent package - 1 dependent repositories - 84 downloads last month - 0 stars on GitHub - 1 maintainer
pyrddlgym-jax 0.1
pyRDDLGym-jax: JAX compilation of RDDL description files, and a differentiable planner in JAX.1 version - Latest release: 3 months ago - 13 downloads last month - 0 stars on GitHub - 1 maintainer
jax-dataloader 0.1.0
Dataloader for jax6 versions - Latest release: 3 months ago - 2 dependent packages - 1 dependent repositories - 234 downloads last month - 32 stars on GitHub - 1 maintainer
Top 8.1% on pypi.org
42 versions - Latest release: 4 months ago - 8 dependent packages - 1 dependent repositories - 2.48 thousand downloads last month - 155 stars on GitHub - 1 maintainer
e3nn-jax 0.20.6
Equivariant convolutional neural networks for the group E(3) of 3 dimensional rotations, translat...42 versions - Latest release: 4 months ago - 8 dependent packages - 1 dependent repositories - 2.48 thousand downloads last month - 155 stars on GitHub - 1 maintainer
lagrangebench 0.1.2
LagrangeBench: A Lagrangian Fluid Mechanics Benchmarking Suite5 versions - Latest release: 4 months ago - 26 downloads last month - 49 stars on GitHub - 2 maintainers
alphapulldown 1.0.4
Pipeline allows massive screening using alphafold59 versions - Latest release: 5 months ago - 865 downloads last month - 165 stars on GitHub - 1 maintainer
moss-rl 0.0.5
A Python library for Reinforcement Learning.6 versions - Latest release: 5 months ago - 33 downloads last month - 2 stars on GitHub - 1 maintainer
dopamax 0.2.0
Reinforcement learning in pure JAX.2 versions - Latest release: 5 months ago - 17 downloads last month - 9 stars on GitHub - 1 maintainer
Top 7.8% on pypi.org
5 versions - Latest release: 5 months ago - 1 dependent repositories - 1.53 thousand downloads last month - 1,541 stars on GitHub - 4 maintainers
colabfold 1.5.5
Making protein folding accessible to all. Predict proteins structures both in google colab and on...5 versions - Latest release: 5 months ago - 1 dependent repositories - 1.53 thousand downloads last month - 1,541 stars on GitHub - 4 maintainers
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: 5 months ago - 1 dependent package - 4 dependent repositories - 62.5 thousand downloads last month - 1 stars on GitHub - 4 maintainers
bayes-jones 0.0.1
Bayesian inference of Jones matrices.1 version - Latest release: 5 months ago - 16 downloads last month - 0 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
deepqmc 1.1.2
Deep-learning quantum Monte Carlo for electrons in real space10 versions - Latest release: 6 months ago - 1 dependent repositories - 172 downloads last month - 321 stars on GitHub - 2 maintainers
xpag 0.2.9
xpag: Exploring Agents18 versions - Latest release: 6 months ago - 82 downloads last month - 20 stars on GitHub - 1 maintainer
femr-oldcpu 0.1.16
Framework for Electronic Medical Records. A python package for building models using EHR data.13 versions - Latest release: 7 months ago - 341 downloads last month - 81 stars on GitHub - 1 maintainer
femr-cuda 0.1.16
Framework for Electronic Medical Records. A python package for building models using EHR data.17 versions - Latest release: 7 months ago - 1 dependent repositories - 374 downloads last month - 81 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
lagrangebench2 0.1.11 removed
LagrangeBench: A Lagrangian Fluid Mechanics Benchmarking Suite1 version - Latest release: 7 months ago - 1 maintainer
haiku-geometric 0.0.5
A collection of graph neural networks implementations in JAX7 versions - Latest release: 8 months ago - 1 dependent repositories - 48 downloads last month - 28 stars on GitHub - 1 maintainer
Top 2.5% on pypi.org
124 versions - Latest release: 8 months ago - 3 dependent packages - 97 dependent repositories - 17.4 thousand downloads last month - 1,304 stars on GitHub - 2 maintainers
underthesea 6.8.0 💰
Vietnamese NLP Toolkit124 versions - Latest release: 8 months ago - 3 dependent packages - 97 dependent repositories - 17.4 thousand downloads last month - 1,304 stars on GitHub - 2 maintainers
non-param-score-est 1.0.0
Non parametric score function estimation library2 versions - Latest release: 8 months ago - 17 downloads last month - 0 stars on GitHub - 1 maintainer
tsuite 1.0
tsuite: Get your RL agent fixed today!1 version - Latest release: 9 months ago - 11 downloads last month - 3 stars on GitHub - 1 maintainer
lagrangebench-dev 0.0.1 removed
LagrangeBench: A Lagrangian Fluid Mechanics Benchmarking Suite1 version - Latest release: 9 months ago - 1 maintainer
lagrangebench-test 0.1.1 removed
LagrangeBench: A Lagrangian Fluid Mechanics Benchmarking Suite1 version - Latest release: 9 months ago - 1 maintainer
Top 3.8% on pypi.org
38 versions - Latest release: 10 months ago - 6 dependent packages - 8 dependent repositories - 3.58 thousand downloads last month - 1,093 stars on GitHub - 1 maintainer
jax-md 0.2.8
Differentiable, Hardware Accelerated, Molecular Dynamics38 versions - Latest release: 10 months ago - 6 dependent packages - 8 dependent repositories - 3.58 thousand downloads last month - 1,093 stars on GitHub - 1 maintainer
segnn-jax 0.7
Steerable E(3) GNN in jax5 versions - Latest release: 10 months ago - 29 downloads last month - 16 stars on GitHub - 1 maintainer
fedjax 0.0.17
Federated learning simulation with JAX.16 versions - Latest release: 11 months ago - 1 dependent repositories - 220 downloads last month - 249 stars on GitHub - 3 maintainers
jax-dips 0.2.1 💰
Differentiable 3D interfacial PDE solvers written in JAX using the Neural Bootstrapping Method.2 versions - Latest release: 11 months ago - 18 downloads last month - 35 stars on GitHub - 1 maintainer
synecdoche 0.0.5
Synecdoche: Hypernetworks for Haiku in JAX5 versions - Latest release: 11 months ago - 48 downloads last month - 2 stars on GitHub - 1 maintainer
modularbayes 0.1.4
Modular Bayesian Inference.3 versions - Latest release: 11 months ago - 26 downloads last month - 1 stars on GitHub - 1 maintainer
chemtrain 0.0.1
Training molecular dynamics potentials.1 version - Latest release: 11 months ago - 23 downloads last month - 14 stars on GitHub - 1 maintainer
egnn-jax 0.3
E(3) GNN in jax2 versions - Latest release: 12 months ago - 20 downloads last month - 6 stars on GitHub - 1 maintainer
explainax 0.0.1
JAX-based Model Explanation and Interpretation Library1 version - Latest release: about 1 year ago - 15 downloads last month - 0 stars on GitHub - 1 maintainer
wax-ml 0.6.4
A Python library for machine-learning and feedback loops on streaming data13 versions - Latest release: about 1 year ago - 1 dependent repositories - 29 downloads last month - 57 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 - 69 downloads last month - 1 maintainer
causal-discovery 0.1.1
2 versions - Latest release: over 1 year ago - 2 downloads last month - 1 maintainercoax 0.1.13
Plug-n-play reinforcement learning with Gymnasium and JAX14 versions - Latest release: over 1 year ago - 1 dependent package - 1 dependent repositories - 220 downloads last month - 159 stars on GitHub - 1 maintainer
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
muax 0.0.2
A library written in Jax that provides help for using DeepMind's mctx on gym-style environments.16 versions - Latest release: over 1 year ago - 161 downloads last month - 40 stars on GitHub - 1 maintainer
cfnet 0.0.12
A counterfactual explanation library using Jax12 versions - Latest release: over 1 year ago - 34 downloads last month - 1 stars on GitHub - 1 maintainer
jax-rex 0.0.11 removed
Jax-based Recourse Explanation Library1 version - Latest release: over 1 year ago - 33 downloads last month - 0 stars on GitHub
dmol-book 1.3.2
Style and Imports for dmol Book13 versions - Latest release: over 1 year ago - 1 dependent repositories - 299 downloads last month - 576 stars on GitHub - 1 maintainer
wazy 0.12.1
Pretrained Bayesian Optimization of Sequences8 versions - Latest release: over 1 year ago - 189 downloads last month - 52 stars on GitHub - 1 maintainer
haiku-mup 0.1.5
A simple port of μP to Haiku/JAX.5 versions - Latest release: over 1 year ago - 1 dependent repositories - 41 downloads last month - 24 stars on GitHub - 1 maintainer