PyRank
  • Insights
  • PyPI
  • GitHub
  • Search
  • Compare
  • Advisories
  • Ecosystem
  • About

Equinox Python Packages

Python packages with the GitHub topic equinox. Sorted by relevance, with stars and monthly downloads.
patrick-kidger
equinox

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

1.7M 3K 194
google
lineax

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

311K 545 38
patrick-kidger
optimistix

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

217K 580 48
patrick-kidger
diffrax

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

189K 2K 180
Ceyron
exponax

Efficient Differentiable n-d PDE Solvers in JAX.

16K 200 17
Ceyron
pdequinox

Neural Emulator Architectures in JAX.

16K 24 0
Ceyron
trainax

Neural Emulator Architectures in JAX.

15K 24 0
nstarman
quax

Multiple dispatch over abstract array types in JAX.

8K 141 7
paganpasta
eqxvision

Root package info.

3K 111 15
patrick-kidger
esm2quinox

An implementation of ESM2 in Equinox+JAX

2K 36 4
GalacticDynamics
diffraxtra

diffrax extras: OOP and vectorization

697 6 3
homerjed
sbgm

Score-based Diffusion models in JAX.

627 18 1
GalacticDynamics
oncequinox

micro-library for making Equinox singletons

398 1 2
RunnersNum40
lerax

Reinforcement Learning in Jax

243 6 0
tttc3
mccube

Markov Chain Cubature via JAX. Efficient SDE solving and Bayesian inference.

238 4 0
gerlero
parametrix

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

163 0 0
camail-official
discretax

Discretax is a light weight collection of state space models implemented in JAX ⚡️

160 38 5
mthiboust
somap

A flexible, fast and scalable python library for Self-Organizing Maps

148 16 1
camail-official
linax

Discretax is a light weight collection of state space models implemented in JAX ⚡️

92 38 5
mlprt
feedbax

Optimal feedback control + interventions, in JAX.

79 6 1
bahremsd
katmer

katmer is a powerful library for optimizing the design of optical thin films using automatic differentiation via JAX and Equinox, enabling efficient and accurate inverse design solutions.

78 11 3
gerlero
frontx

⚛️ Nonlinear diffusion problems with JAX

76 2 0
    • Data from PyPI, GitHub, ClickHouse, and BigQuery