├── .gitignore ├── LICENSE ├── README.md ├── examples ├── anil-evolution.py ├── cavia-conjugate-gradient.py ├── compsyn-eqprob.py ├── gainmod-reptile.py └── maml-omniglot.py ├── metax ├── __init__.py ├── data │ ├── __init__.py │ ├── base.py │ ├── dataset │ │ ├── __init__.py │ │ ├── base.py │ │ ├── family.py │ │ └── sinusoid.py │ ├── synthetic.py │ └── utils.py ├── energy │ ├── __init__.py │ ├── base.py │ ├── loss.py │ ├── metrics.py │ └── regularizer.py ├── learner │ ├── __init__.py │ ├── base.py │ ├── eqprop.py │ ├── evolution.py │ ├── implicit.py │ ├── maml.py │ └── reptile.py ├── models │ ├── __init__.py │ ├── conv.py │ ├── linear.py │ └── mlp.py ├── module │ ├── __init__.py │ ├── anil.py │ ├── base.py │ ├── cavia.py │ ├── compsyn.py │ ├── gainmod.py │ └── init.py └── utils │ ├── __init__.py │ ├── pytree.py │ └── utils.py ├── setup.py └── tests ├── data ├── __init__.py └── datasets │ ├── __init__.py │ ├── test_family.py │ └── test_sinusoid.py ├── meta ├── __init__.py └── test_learner.py └── utils ├── __init__.py └── test_pytree.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smonsays/metax/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smonsays/metax/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smonsays/metax/HEAD/README.md -------------------------------------------------------------------------------- /examples/anil-evolution.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smonsays/metax/HEAD/examples/anil-evolution.py -------------------------------------------------------------------------------- /examples/cavia-conjugate-gradient.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smonsays/metax/HEAD/examples/cavia-conjugate-gradient.py -------------------------------------------------------------------------------- /examples/compsyn-eqprob.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smonsays/metax/HEAD/examples/compsyn-eqprob.py -------------------------------------------------------------------------------- /examples/gainmod-reptile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smonsays/metax/HEAD/examples/gainmod-reptile.py -------------------------------------------------------------------------------- /examples/maml-omniglot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smonsays/metax/HEAD/examples/maml-omniglot.py -------------------------------------------------------------------------------- /metax/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smonsays/metax/HEAD/metax/__init__.py -------------------------------------------------------------------------------- /metax/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smonsays/metax/HEAD/metax/data/__init__.py -------------------------------------------------------------------------------- /metax/data/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smonsays/metax/HEAD/metax/data/base.py -------------------------------------------------------------------------------- /metax/data/dataset/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smonsays/metax/HEAD/metax/data/dataset/__init__.py -------------------------------------------------------------------------------- /metax/data/dataset/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smonsays/metax/HEAD/metax/data/dataset/base.py -------------------------------------------------------------------------------- /metax/data/dataset/family.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smonsays/metax/HEAD/metax/data/dataset/family.py -------------------------------------------------------------------------------- /metax/data/dataset/sinusoid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smonsays/metax/HEAD/metax/data/dataset/sinusoid.py -------------------------------------------------------------------------------- /metax/data/synthetic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smonsays/metax/HEAD/metax/data/synthetic.py -------------------------------------------------------------------------------- /metax/data/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smonsays/metax/HEAD/metax/data/utils.py -------------------------------------------------------------------------------- /metax/energy/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smonsays/metax/HEAD/metax/energy/__init__.py -------------------------------------------------------------------------------- /metax/energy/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smonsays/metax/HEAD/metax/energy/base.py -------------------------------------------------------------------------------- /metax/energy/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smonsays/metax/HEAD/metax/energy/loss.py -------------------------------------------------------------------------------- /metax/energy/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smonsays/metax/HEAD/metax/energy/metrics.py -------------------------------------------------------------------------------- /metax/energy/regularizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smonsays/metax/HEAD/metax/energy/regularizer.py -------------------------------------------------------------------------------- /metax/learner/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smonsays/metax/HEAD/metax/learner/__init__.py -------------------------------------------------------------------------------- /metax/learner/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smonsays/metax/HEAD/metax/learner/base.py -------------------------------------------------------------------------------- /metax/learner/eqprop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smonsays/metax/HEAD/metax/learner/eqprop.py -------------------------------------------------------------------------------- /metax/learner/evolution.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smonsays/metax/HEAD/metax/learner/evolution.py -------------------------------------------------------------------------------- /metax/learner/implicit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smonsays/metax/HEAD/metax/learner/implicit.py -------------------------------------------------------------------------------- /metax/learner/maml.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smonsays/metax/HEAD/metax/learner/maml.py -------------------------------------------------------------------------------- /metax/learner/reptile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smonsays/metax/HEAD/metax/learner/reptile.py -------------------------------------------------------------------------------- /metax/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smonsays/metax/HEAD/metax/models/__init__.py -------------------------------------------------------------------------------- /metax/models/conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smonsays/metax/HEAD/metax/models/conv.py -------------------------------------------------------------------------------- /metax/models/linear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smonsays/metax/HEAD/metax/models/linear.py -------------------------------------------------------------------------------- /metax/models/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smonsays/metax/HEAD/metax/models/mlp.py -------------------------------------------------------------------------------- /metax/module/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smonsays/metax/HEAD/metax/module/__init__.py -------------------------------------------------------------------------------- /metax/module/anil.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smonsays/metax/HEAD/metax/module/anil.py -------------------------------------------------------------------------------- /metax/module/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smonsays/metax/HEAD/metax/module/base.py -------------------------------------------------------------------------------- /metax/module/cavia.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smonsays/metax/HEAD/metax/module/cavia.py -------------------------------------------------------------------------------- /metax/module/compsyn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smonsays/metax/HEAD/metax/module/compsyn.py -------------------------------------------------------------------------------- /metax/module/gainmod.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smonsays/metax/HEAD/metax/module/gainmod.py -------------------------------------------------------------------------------- /metax/module/init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smonsays/metax/HEAD/metax/module/init.py -------------------------------------------------------------------------------- /metax/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smonsays/metax/HEAD/metax/utils/__init__.py -------------------------------------------------------------------------------- /metax/utils/pytree.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smonsays/metax/HEAD/metax/utils/pytree.py -------------------------------------------------------------------------------- /metax/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smonsays/metax/HEAD/metax/utils/utils.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smonsays/metax/HEAD/setup.py -------------------------------------------------------------------------------- /tests/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/data/datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/data/datasets/test_family.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smonsays/metax/HEAD/tests/data/datasets/test_family.py -------------------------------------------------------------------------------- /tests/data/datasets/test_sinusoid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smonsays/metax/HEAD/tests/data/datasets/test_sinusoid.py -------------------------------------------------------------------------------- /tests/meta/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/meta/test_learner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smonsays/metax/HEAD/tests/meta/test_learner.py -------------------------------------------------------------------------------- /tests/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/utils/test_pytree.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/smonsays/metax/HEAD/tests/utils/test_pytree.py --------------------------------------------------------------------------------