Fabian Pedregosa

TODO: - [x] Frame it as a bi-level problem, as the implicit maml paper (https://arxiv.org/pdf/1909.04630.pdf) - [x] Move to implicit diff directory

I think this is finally ready for review 🎉 !

one can see the notebook here: https://github.com/google/jaxopt/blob/d7fc1d5407c81ab3c70bb09b818101fbcb4e1e3e/docs/notebooks/implicit_diff/maml.ipynb

There are still a couple of issues left, but I believe these can be tackled after merge: 1. The notebook takes a while to run, like a couple of hours...

@asteroidhouse might also be interested by this 😄

thanks for the review! Implemented the suggestions. Please add the pull-ready label if you're satisfied :-)

Thanks @Algue-Rythme ! That almost works. Unfortunately, to make it work in `flax_image_classif.py` I need also to remove the pre_update=print_accuracy argument of the solver, as otherwise it crashes with this...

I tried to protect the problmatic lines with a `with jax.disable_jit():` statement but it still failed

Solved for the flax_resnet.py example in #119 . Leaving this issue open since there are other examples where the GPU utilization is poor

yes!