Awesome List

A list of community projects that use Equinox and its ecosystem! Feel free to write a PR to add your own projects so that the community may benefit from as many examples as possible.

Models

Mistral 7B: GitHub
Implementation of the Mistral 7B model.

PaLM: GitHub
Implementation of the Pathways Language Model (PaLM).

HiFiGAN: GitHub
HiFiGAN vocoder using two discriminators to map mel-spectrograms to audio.

nanoGPT: GitHub
Port of nanoGPT, a minimalist implementation of a Generative Pre-trained Transformer.

VQVAE: GitHub
Implementation of a VQ-VAE (Vector Quantized Variational Autoencoder) for audio generation.

ESM2quinox: GitHub
The ESM2 protein language model.

Libraries

Levanter: GitHub
Deep learning library for training large language models.

Eqxvision: GitHub
Computer vision library built on Equinox, inspired by torchvision.

Haliax: GitHub
Distributed multi-dimensional arrays, designed for large-scale machine learning.

Diffrax: GitHub
Numerical differential equation solvers.

Optimistix: GitHub
Nonlinear optimisation: root finding, fixed point, minimisation, least squares.

Lineax: GitHub
Linear solvers, linear operators, and linear least squares.

Quax: GitHub
Multiple dispatch (over abstract array types).

sympy2jax: GitHub
Converts sympy expressions into JAX expressions.

quadax: GitHub
Numerical quadrature.

traceax: GitHub
Stochastic trace estimation.

Dynamiqs: GitHub
High-performance quantum systems simulation with JAX.

unxt: GitHub
Unitful quantities.

coordinax: GitHub
Coordinates (like astropy.coordinates).

galax: GitHub
Galactic and gravitational dynamics.

FlowJAX: GitHub
Normalizing flows and probabilistic machine learning.

statedict2pytree: GitHub
Converts PyTorch state_dicts into Equinox pytrees.