LICENSE
MANIFEST.in
README.md
setup.py
jax_meta/__init__.py
jax_meta/version.py
jax_meta.egg-info/PKG-INFO
jax_meta.egg-info/SOURCES.txt
jax_meta.egg-info/dependency_links.txt
jax_meta.egg-info/requires.txt
jax_meta.egg-info/top_level.txt
jax_meta/datasets/__init__.py
jax_meta/datasets/base.py
jax_meta/datasets/cifarfs.py
jax_meta/datasets/fc100.py
jax_meta/datasets/leo.py
jax_meta/datasets/miniimagenet.py
jax_meta/datasets/omniglot.py
jax_meta/datasets/tieredimagenet.py
jax_meta/datasets/assets/repository.json
jax_meta/datasets/assets/cifar100/cifar-fs/test.json
jax_meta/datasets/assets/cifar100/cifar-fs/train.json
jax_meta/datasets/assets/cifar100/cifar-fs/val.json
jax_meta/datasets/assets/cifar100/fc100/test.json
jax_meta/datasets/assets/cifar100/fc100/train.json
jax_meta/datasets/assets/cifar100/fc100/val.json
jax_meta/datasets/assets/cub/test.json
jax_meta/datasets/assets/cub/train.json
jax_meta/datasets/assets/cub/val.json
jax_meta/datasets/assets/doublemnist/test.json
jax_meta/datasets/assets/doublemnist/train.json
jax_meta/datasets/assets/doublemnist/val.json
jax_meta/datasets/assets/omniglot/test.json
jax_meta/datasets/assets/omniglot/train.json
jax_meta/datasets/assets/omniglot/val.json
jax_meta/datasets/assets/triplemnist/test.json
jax_meta/datasets/assets/triplemnist/train.json
jax_meta/datasets/assets/triplemnist/val.json
jax_meta/datasets/transforms/__init__.py
jax_meta/datasets/transforms/functional.py
jax_meta/datasets/transforms/test_functional.py
jax_meta/metalearners/__init__.py
jax_meta/metalearners/anil.py
jax_meta/metalearners/base.py
jax_meta/metalearners/imaml.py
jax_meta/metalearners/lrd2.py
jax_meta/metalearners/maml.py
jax_meta/metalearners/meta_sgd.py
jax_meta/metalearners/r2d2.py
jax_meta/modules/__init__.py
jax_meta/modules/conv.py
jax_meta/modules/r2d2.py
jax_meta/modules/relation_network.py
jax_meta/modules/resnet.py
jax_meta/utils/__init__.py
jax_meta/utils/arguments.py
jax_meta/utils/data.py
jax_meta/utils/io.py
jax_meta/utils/losses.py
jax_meta/utils/metrics.py
jax_meta/utils/wandb_utils.py
tests/metalearners/__init__.py
tests/metalearners/test_maml.py