Skip to content

Commit aa57716

Browse files
Added external repos in Examples (#947)
* Added external repos in Examples * Docs awesome list
1 parent d08de05 commit aa57716

File tree

3 files changed

+39
-10
lines changed

3 files changed

+39
-10
lines changed

examples/awesome-list.md

Lines changed: 32 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,32 @@
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.

examples/mistral_7b.md

Lines changed: 0 additions & 3 deletions
This file was deleted.

mkdocs.yml

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -103,19 +103,19 @@ nav:
103103
- Introductory:
104104
- CNN on MNIST: 'examples/mnist.ipynb'
105105
- Train RNN: 'examples/train_rnn.ipynb'
106-
- Advanced:
107-
- Generative score-based diffusion: 'examples/score_based_diffusion.ipynb'
108-
- BERT language model: 'examples/bert.ipynb'
109-
- U-Net implementation: 'examples/unet.ipynb'
110-
- Vision transformer: 'examples/vision_transformer.ipynb'
111-
- Image GAN: 'examples/deep_convolutional_gan.ipynb'
112-
- Mistral 7B: 'examples/mistral_7b.md'
113106
- Features:
114107
- Freezing parameters: 'examples/frozen_layer.ipynb'
115108
- Compatibility with init-apply libraries: 'examples/init_apply.ipynb'
116109
- Stateful operations (e.g. BatchNorm): 'examples/stateful.ipynb'
117110
- Autoparallelism (e.g. multi-GPU): 'examples/parallelism.ipynb'
118111
- Serialisation (with hyperparameters): 'examples/serialisation.ipynb'
112+
- Advanced (Notebooks):
113+
- Generative score-based diffusion: 'examples/score_based_diffusion.ipynb'
114+
- BERT language model: 'examples/bert.ipynb'
115+
- U-Net implementation: 'examples/unet.ipynb'
116+
- Vision transformer: 'examples/vision_transformer.ipynb'
117+
- Image GAN: 'examples/deep_convolutional_gan.ipynb'
118+
- Awesome list: 'awesome-list.md'
119119
- Basic API:
120120
- Modules:
121121
- 'api/module/module.md'

0 commit comments

Comments
 (0)