PyPI Stats

Search Packages

Find Python packages by name, description, GitHub topic, or filter by metrics
patrick-kidger
equinox

Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/

1.1M 3K 194
google
lineax

Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax

272K 541 38
patrick-kidger
optimistix

Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/

208K 575 48
patrick-kidger
diffrax

Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/

180K 2K 178
Ceyron
exponax

Efficient Differentiable n-d PDE Solvers in JAX.

13K 199 17
Ceyron
pdequinox

Neural Emulator Architectures in JAX.

13K 24 0
Ceyron
trainax

Neural Emulator Architectures in JAX.

13K 24 0
nstarman
quax

Multiple dispatch over abstract array types in JAX.

9K 141 7
patrick-kidger
esm2quinox

An implementation of ESM2 in Equinox+JAX

2K 36 4
paganpasta
eqxvision

A Python package of computer vision models for the Equinox ecosystem.

2K 111 15
homerjed
sbgm

Score-based Diffusion models in JAX.

1K 18 1
GalacticDynamics
diffraxtra

Extras for Diffrax: OOP and vectorization

1K 6 3
RunnersNum40
lerax

Deep Reinforcement Learning with JAX and Equinox.

361 6 0
gerlero
parametrix

🪄 Flax-like computed parameters for bare JAX (and Equinox)

284 0 0
GalacticDynamics
oncequinox

Create singleton Equinox modules.

270 1 2
tttc3
mccube

Markov chain cubature via JAX.

247 4 0
mthiboust
somap

Self-Organizing Maps in python

163 16 1
camail-official
discretax

State Space Models in Jax

126 37 5
mlprt
feedbax

Optimal feedback control + interventions, in JAX.

74 6 1
camail-official
linax

State Space Models in Jax

71 34 5
gerlero
frontx

⚛️ Nonlinear diffusion problems with JAX

70 2 0
bahremsd
katmer

project_description

68 11 3