|
| 1 | +# Awesome List |
| 2 | + |
| 3 | +A curated list of external projects from the community that leverage the Equinox framework to build models using JAX. |
| 4 | + |
| 5 | +Feel free to create a PR to add your own projects so that the community may benefit from as many examples as possible ! |
| 6 | +## Advanced (External Repos) |
| 7 | +- **Mistral 7B**: [GitHub](https://github.com/AakashKumarNain/mistral_jax) |
| 8 | + - Implementation of the Mistral 7B model. |
| 9 | +- **PaLM**: [GitHub](https://github.com/lucidrains/PaLM-jax) |
| 10 | + - Implementation of the Pathways Language Model (PaLM). |
| 11 | +- **HiFiGAN**: [GitHub](https://github.com/TugdualKerjan/hifigan-jax) |
| 12 | + - HiFiGAN vocoder using two discriminators to map mel-spectrograms to audio. |
| 13 | +- **nanoGPT**: [GitHub](https://github.com/TugdualKerjan/nanoEQXGPT) |
| 14 | + - Port of nanoGPT, a minimalist implementation of a Generative Pre-trained Transformer. |
| 15 | +- **VQVAE**: [GitHub](https://github.com/TugdualKerjan/audio-vae-jax) |
| 16 | + - Implementation of a VQ-VAE (Vector Quantized Variational Autoencoder) for audio generation. |
| 17 | + |
| 18 | +## Projects Based on Equinox |
| 19 | +- **Levanter**: [GitHub](https://github.com/stanford-crfm/levanter) |
| 20 | + - Deep learning library for training large language models. |
| 21 | +- **Eqxvision**: [GitHub](https://github.com/paganpasta/eqxvision) |
| 22 | + - Computer vision library built on Equinox, inspired by torchvision. |
| 23 | +- **Haliax**: [GitHub](https://github.com/stanford-crfm/haliax) |
| 24 | + - Library for distributed multi-dimensional arrays, designed for large-scale machine learning. |
| 25 | +- **Diffrax**: [GitHub](https://github.com/patrick-kidger/diffrax) |
| 26 | + - Library for numerical differential equation solvers. |
| 27 | +- **Optimistix**: [GitHub](https://github.com/patrick-kidger/optimistix) |
| 28 | + - Library for nonlinear optimization. |
| 29 | +- **FlowJAX**: [GitHub](https://github.com/danielward27/flowjax?tab=readme-ov-file) |
| 30 | + - Library for normalizing flows and probabilistic machine learning. |
| 31 | +- **Lineax**: [GitHub](https://github.com/patrick-kidger/lineax) |
| 32 | + - Library for linear solvers, linear least squares, and eigenvalue solvers. |
0 commit comments