Metax: Meta-learning extensions for JAX This is an experimental library Metax is a collection of extensions for meta-learning in JAX. Metax contains a subset of the utility functions from Torchmeta.