tristandeleu / metax

A collection of extensions for meta-learning in JAX

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.