[JAX] Meta Learning Tutorial?

[ImahnShekhzadeh]

Hi Phillip,

Thank you for open-sourcing the course!

I was wondering whether there is a specific reason that the meta learning tutorial was not "ported" to Jax? :slightly_smiling_face: Is it because you did not find the time, or rather because of issues you encountered while doing the porting?

[phlippe]

Hi Imahn,

There is no specific reason for the meta-learning tutorial being not yet in JAX, I simply didn't find the time to do it yet. I hope to get to it at some point, since by using the jax.grad transformations, it is actually easier to implement these algorithms in JAX than in PyTorch :)