Implementation of 'Learning to learn by gradient descent by gradient descent' (https://arxiv.org/abs/1606.04474) in JAX(https://github.com/google/jax) and dm-haiku(https://github.com/deepmind/dm-haiku).
Tested on Quadratic Problem(implementations/quadratic_problem) and Classifying MNIST Dataset with MLP((implementations/mlp).