Skip to content

Commit 1b507b9

Browse files
Update awesome list with more things
1 parent aa57716 commit 1b507b9

File tree

3 files changed

+79
-38
lines changed

3 files changed

+79
-38
lines changed

docs/awesome-list.md

+73
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,73 @@
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.

examples/awesome-list.md

-32
This file was deleted.

mkdocs.yml

+6-6
Original file line numberDiff line numberDiff line change
@@ -103,18 +103,18 @@ 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'
106112
- Features:
107113
- Freezing parameters: 'examples/frozen_layer.ipynb'
108114
- Compatibility with init-apply libraries: 'examples/init_apply.ipynb'
109115
- Stateful operations (e.g. BatchNorm): 'examples/stateful.ipynb'
110116
- Autoparallelism (e.g. multi-GPU): 'examples/parallelism.ipynb'
111117
- 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'
118118
- Awesome list: 'awesome-list.md'
119119
- Basic API:
120120
- Modules:

0 commit comments

Comments
 (0)