Update README.md

README.md

@@ -21,12 +21,13 @@ This is a landing page for **Gemma Scope**, a comprehensive, open suite of spars
 - https://huggingface.co/google/gemma-scope-9b-it-res
 - https://huggingface.co/google/gemma-scope-2b-pt-transcoders
 
-[This tutorial](https://colab.research.google.com/drive/17dQFYUYnuKnP6OwQPH9v_GSYUW5aj-Rp?ts=66a77041) has instructions on how to load the SAEs.
+[This tutorial](https://colab.research.google.com/drive/17dQFYUYnuKnP6OwQPH9v_GSYUW5aj-Rp?ts=66a77041) has instructions on how to load the SAEs, and [this tutorial] explains and implements JumpReLU SAE training in PyTorch and JAX.
 
 # Key links:
 
 ![](gs-demo-tweet.gif)
 - Check out the [interactive Gemma Scope demo](https://www.neuronpedia.org/gemma-scope) made by [Neuronpedia](https://www.neuronpedia.org/).
+- (NEW!) We have a colab notebook tutorial for JumpReLU SAE training in JAX and PyTorch [here](https://colab.research.google.com/drive/1PlFzI_PWGTN9yCQLuBcSuPJUjgHL7GiD).
 - Learn more about Gemma Scope in our [Google DeepMind blog post](https://deepmind.google/discover/blog/gemma-scope-helping-the-safety-community-shed-light-on-the-inner-workings-of-language-models).
 - Check out our [Google Colab notebook tutorial](https://colab.research.google.com/drive/17dQFYUYnuKnP6OwQPH9v_GSYUW5aj-Rp?ts=66a77041) for how to use Gemma Scope.
 - Read [the Gemma Scope technical report](https://arxiv.org/abs/2408.05147).