metax (pypi): A collection of extensions for meta-learning in JAX
Keywords: meta-learning|jax|few-shot|few-shot learning