A collection of extensions for meta-learning in JAX - View it on GitHub