From dd85e2a8e9b2104aa809cdb175317d203f8175db Mon Sep 17 00:00:00 2001 From: Patrick Kidger <33688385+patrick-kidger@users.noreply.github.com> Date: Sun, 12 May 2024 15:14:48 +0200 Subject: [PATCH] Added Mistral 7B example --- examples/mistral_7b.md | 3 +++ mkdocs.yml | 1 + 2 files changed, 4 insertions(+) create mode 100644 examples/mistral_7b.md diff --git a/examples/mistral_7b.md b/examples/mistral_7b.md new file mode 100644 index 00000000..72df0e72 --- /dev/null +++ b/examples/mistral_7b.md @@ -0,0 +1,3 @@ +# Mistral 7B + +An Equinox implementation of [Mistral 7B](https://github.com/mistralai/mistral-src/) model is available in the [AakashKumarNain/mistral_jax](https://github.com/AakashKumarNain/mistral_jax) repository! diff --git a/mkdocs.yml b/mkdocs.yml index bf5bda76..68331eff 100644 --- a/mkdocs.yml +++ b/mkdocs.yml @@ -109,6 +109,7 @@ nav: - U-Net implementation: 'examples/unet.ipynb' - Vision transformer: 'examples/vision_transformer.ipynb' - Image GAN: 'examples/deep_convolutional_gan.ipynb' + - Mistral 7B: 'examples/mistral_7b.md' - Features: - Freezing parameters: 'examples/frozen_layer.ipynb' - Compatibility with init-apply libraries: 'examples/init_apply.ipynb'