|
| 1 | +# Awesome List |
| 2 | + |
| 3 | +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. |
| 4 | + |
| 5 | +## Models |
| 6 | + |
| 7 | +**Mistral 7B**: [GitHub](https://github.com/AakashKumarNain/mistral_jax) |
| 8 | +Implementation of the Mistral 7B model. |
| 9 | + |
| 10 | +**PaLM**: [GitHub](https://github.com/lucidrains/PaLM-jax) |
| 11 | +Implementation of the Pathways Language Model (PaLM). |
| 12 | + |
| 13 | +**HiFiGAN**: [GitHub](https://github.com/TugdualKerjan/hifigan-jax) |
| 14 | +HiFiGAN vocoder using two discriminators to map mel-spectrograms to audio. |
| 15 | + |
| 16 | +**nanoGPT**: [GitHub](https://github.com/TugdualKerjan/nanoEQXGPT) |
| 17 | +Port of nanoGPT, a minimalist implementation of a Generative Pre-trained Transformer. |
| 18 | + |
| 19 | +**VQVAE**: [GitHub](https://github.com/TugdualKerjan/audio-vae-jax) |
| 20 | +Implementation of a VQ-VAE (Vector Quantized Variational Autoencoder) for audio generation. |
| 21 | + |
| 22 | +**ESM2quinox**: [GitHub](https://github.com/patrick-kidger/esm2quinox) |
| 23 | +The ESM2 protein language model. |
| 24 | + |
| 25 | +## Libraries |
| 26 | + |
| 27 | +**Levanter**: [GitHub](https://github.com/stanford-crfm/levanter) |
| 28 | +Deep learning library for training large language models. |
| 29 | + |
| 30 | +**Eqxvision**: [GitHub](https://github.com/paganpasta/eqxvision) |
| 31 | +Computer vision library built on Equinox, inspired by torchvision. |
| 32 | + |
| 33 | +**Haliax**: [GitHub](https://github.com/stanford-crfm/haliax) |
| 34 | +Distributed multi-dimensional arrays, designed for large-scale machine learning. |
| 35 | + |
| 36 | +**Diffrax**: [GitHub](https://github.com/patrick-kidger/diffrax) |
| 37 | +Numerical differential equation solvers. |
| 38 | + |
| 39 | +**Optimistix**: [GitHub](https://github.com/patrick-kidger/optimistix) |
| 40 | +Nonlinear optimisation: root finding, fixed point, minimisation, least squares. |
| 41 | + |
| 42 | +**Lineax**: [GitHub](https://github.com/patrick-kidger/lineax) |
| 43 | +Linear solvers, linear operators, and linear least squares. |
| 44 | + |
| 45 | +**Quax**: [GitHub](https://github.com/patrick-kidger/quax) |
| 46 | +Multiple dispatch (over abstract array types). |
| 47 | + |
| 48 | +**sympy2jax**: [GitHub](https://github.com/patrick-kidger/sympy2jax) |
| 49 | +Converts sympy expressions into JAX expressions. |
| 50 | + |
| 51 | +**quadax**: [GitHub](https://github.com/f0uriest/quadax) |
| 52 | +Numerical quadrature. |
| 53 | + |
| 54 | +**traceax**: [GitHub](https://github.com/mancusolab/traceax) |
| 55 | +Stochastic trace estimation. |
| 56 | + |
| 57 | +**Dynamiqs:** [GitHub](https://github.com/dynamiqs/dynamiqs) |
| 58 | +High-performance quantum systems simulation with JAX. |
| 59 | + |
| 60 | +**unxt**: [GitHub](https://github.com/GalacticDynamics/unxt) |
| 61 | +Unitful quantities. |
| 62 | + |
| 63 | +**coordinax**: [GitHub](https://github.com/GalacticDynamics/coordinax) |
| 64 | +Coordinates (like `astropy.coordinates`). |
| 65 | + |
| 66 | +**galax**: [GitHub](https://github.com/GalacticDynamics/galax) |
| 67 | +Galactic and gravitational dynamics. |
| 68 | + |
| 69 | +**FlowJAX**: [GitHub](https://github.com/danielward27/flowjax) |
| 70 | +Normalizing flows and probabilistic machine learning. |
| 71 | + |
| 72 | +**statedict2pytree**: [GitHub](https://github.com/Artur-Galstyan/statedict2pytree) |
| 73 | +Converts PyTorch `state_dict`s into Equinox pytrees. |
0 commit comments