Blog post: Learning JAX by Learning to Learn

Follow the full discussion on Reddit.
I recently published a new blog post, which goes over how meta-learned optimizers work and how to implement them in JAX. JAX's composable function transforms make implementing meta-learning algorithms very straightforward. If you're interested in JAX or meta learning give it a read!