pypi.org "equinox" keyword
View the packages on the pypi.org package registry that are tagged with the "equinox" keyword.
Top 1.8% on pypi.org
58 versions - Latest release: 23 days ago - 78 dependent packages - 26 dependent repositories - 299 thousand downloads last month - 2,093 stars on GitHub - 1 maintainer
equinox 0.12.1 💰
Elegant easy-to-use neural networks in JAX.58 versions - Latest release: 23 days ago - 78 dependent packages - 26 dependent repositories - 299 thousand downloads last month - 2,093 stars on GitHub - 1 maintainer
Top 1.4% on pypi.org
44 versions - Latest release: 18 days ago - 125 dependent packages - 63 dependent repositories - 1.23 million downloads last month - 1,168 stars on GitHub - 1 maintainer
jaxtyping 0.3.1 💰
Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays.44 versions - Latest release: 18 days ago - 125 dependent packages - 63 dependent repositories - 1.23 million downloads last month - 1,168 stars on GitHub - 1 maintainer
cryojax 0.3.4
Cryo-EM image simulation and analysis powered by JAX18 versions - Latest release: 16 days ago - 609 downloads last month - 34 stars on GitHub - 1 maintainer
somap 0.1.1
Self-Organizing Maps in python3 versions - Latest release: 12 months ago - 56 downloads last month - 9 stars on GitHub - 1 maintainer
typinox 0.2.1
Enhanced runtime type-checked JAX modules3 versions - Latest release: 6 days ago - 150 downloads last month - 0 stars on GitHub - 1 maintainer
ezjaxtyping 0.2.20
Type annotations and runtime checking for shape and dtype of JAX arrays, and PyTrees.1 version - Latest release: over 1 year ago - 1 dependent package - 75 downloads last month - 0 stars on GitHub - 1 maintainer
feedbax 0.1.2
Optimal feedback control + interventions in JAX.2 versions - Latest release: about 1 year ago - 70 downloads last month - 4 stars on GitHub - 1 maintainer
optimistix 0.0.10 💰
Nonlinear optimisation in JAX and Equinox.10 versions - Latest release: 4 months ago - 7 dependent packages - 39.6 thousand downloads last month - 394 stars on GitHub - 1 maintainer
quax 0.2.0 💰
Multiple dispatch in JAX via custom interpreters.9 versions - Latest release: 8 days ago - 6 dependent packages - 1.55 thousand downloads last month - 115 stars on GitHub - 1 maintainer
distreqx 0.0.1 💰
Lightweight JAX library of probability distributions and bijectors.1 version - Latest release: 7 months ago - 83 downloads last month - 28 stars on GitHub - 1 maintainer
mccube 0.0.3
Markov chain cubature via JAX.3 versions - Latest release: about 1 year ago - 1 dependent repositories - 111 downloads last month - 2 stars on GitHub - 1 maintainer
Top 2.9% on pypi.org
22 versions - Latest release: about 1 month ago - 26 dependent packages - 9 dependent repositories - 40.4 thousand downloads last month - 1,577 stars on GitHub - 1 maintainer
diffrax 0.7.0 💰
GPU+autodiff-capable ODE/SDE/CDE solvers written in JAX.22 versions - Latest release: about 1 month ago - 26 dependent packages - 9 dependent repositories - 40.4 thousand downloads last month - 1,577 stars on GitHub - 1 maintainer
exponax 0.1.0
Efficient differentiable PDE solvers in JAX.2 versions - Latest release: 6 months ago - 1.13 thousand downloads last month - 24 stars on GitHub - 1 maintainer
trainax 0.0.2
Training methodologies for Autoregressive Neural Emulators in JAX built on top of Equinox & Optax.2 versions - Latest release: 6 months ago - 155 downloads last month - 16 stars on GitHub - 1 maintainer
seasons-cal 1.0.2
3 versions - Latest release: 6 months ago - 86 downloads last month - 0 stars on GitHub - 1 maintainersympy2jax 0.0.6 💰
Turn SymPy expressions into trainable JAX expressions.6 versions - Latest release: 4 months ago - 3 dependent packages - 4.14 thousand downloads last month - 332 stars on GitHub - 1 maintainer
pdequinox 0.1.2
Neural PDE Emulator Architectures in JAX built on top of Equinox.2 versions - Latest release: 6 months ago - 160 downloads last month - 16 stars on GitHub - 1 maintainer
sbgm 0.0.21
Score-based Diffusion models in JAX.11 versions - Latest release: 3 months ago - 300 downloads last month - 5 stars on GitHub - 1 maintainer
eqxvision 0.2.8
Root package info.28 versions - Latest release: about 2 years ago - 802 downloads last month - 104 stars on GitHub - 1 maintainer
Top 9.6% on pypi.org
8 versions - Latest release: 23 days ago - 7 dependent packages - 1 dependent repositories - 45.3 thousand downloads last month - 425 stars on GitHub - 1 maintainer
lineax 0.0.8 💰
Linear solvers in JAX and Equinox.8 versions - Latest release: 23 days ago - 7 dependent packages - 1 dependent repositories - 45.3 thousand downloads last month - 425 stars on GitHub - 1 maintainer
flowjax 17.1.1
Easy to use distributions, bijections and normalizing flows in JAX.46 versions - Latest release: about 2 months ago - 1 dependent package - 1 dependent repositories - 1.46 thousand downloads last month - 151 stars on GitHub - 1 maintainer
jax-optix 0.1.10
Zero-overhead functional lensing for JAX PyTrees11 versions - Latest release: about 2 months ago - 397 downloads last month - 6 stars on GitHub - 1 maintainer
atomcloud 0.0.1
Conventional method for fitting atom cloud, BECs and bimodal atom cloud/BEC distributions in 1D a...1 version - Latest release: almost 2 years ago - 49 downloads last month - 1 stars on GitHub - 1 maintainer
paramax 0.0.0
Parameterizations and parameter constraints for JAX PyTrees.1 version - Latest release: 6 months ago - 531 downloads last month - 15 stars on GitHub - 1 maintainer
diffraxtra 1.5.1
diffrax extras: OOP and vectorization10 versions - Latest release: about 2 months ago - 1.99 thousand downloads last month - 2 stars on GitHub - 2 maintainers
esm2quinox 0.1.0 💰
ESM2 implemented in Equinox+JAX.1 version - Latest release: 2 months ago - 289 downloads last month - 24 stars on GitHub - 1 maintainer
Related Keywords
jax
23
deep-learning
14
neural-networks
14
machine-learning
5
typing
4
python
3
sciml
3
pde
3
neural operator
2
boundary-conditions
2
dimensions
2
neural-differential-equations
2
dynamical-systems
2
diffrax
2
differential-equations
2
probability
2
pytorch
2
python-typing
2
protein
1
diffusion
1
score-based-diffusion
1
generative-models
1
diffusion-models
1
sympy
1
solstice
1
seasons
1
ics
1
icalendar
1
ical
1
calendar
1
cryo-EM
1
electron-microscopy
1
spectral-methods
1
simulation
1
protein-language-model
1
esm2
1
fitting
1
BEC
1
optics
1
lens
1
functional
1
normalizing-flows
1
normalising-flows
1
linear-algebra
1
numerical-methods
1
linear-solvers
1
least-squares
1
vision
1
score-based-generative-models
1
score-based-generative-modeling
1
pde-solver
1
mcmc
1
cubature
1
statistics
1
distributions
1
annotations
1
multiple-dispatch
1
optimisation
1
optimization
1
numerical-optimization
1
nonlinear-programming
1
levenberg-marquardt
1
motor-control
1
computational-neuroscience
1
biomechanics
1
optimal-control
1
optimal-feedback-control
1
fft
1
differentiable-physics
1
etdrk
1
ai
1
bio-inspired
1
self-organizing-map
1
unsupervised-clustering
1
unsupervised-learning
1
type-checking
1
stochastic-differential-equations
1
sdes
1
quadrature
1
probabilistic-programming
1
bayesian-inference
1
stochastic differential equations
1
sampling
1