Implementation of Model-Agnostic Meta-Learning (MAML) in Jax - View it on GitHub