Omniglot MAML examples

In this directory we've provided some examples of training omniglot that reproduce the experiments from the original MAML paper.

They can be run via python {filename}.

maml-omniglot-higher.py uses the facebookresearch/higher metalearning package and is the reference implementation. It runs all of its tasks sequentially.

maml-omniglot-transforms.py uses functorch. It runs all of its tasks in parallel. In theory this should lead to some speedups, but we haven't finished implementing all the rules for vmap that would actually make training faster.

maml-omniglot-ptonly.py is an implementation of maml-omniglot-transforms.py that runs all of its tasks sequentially (and also doesn't use the higher package).