search

phlippe / uvadlc_notebooks

Repository of Jupyter notebook tutorials for teaching the Deep Learning Course at the University of Amsterdam (MSc AI), Fall 2023
https://uvadlc-notebooks.readthedocs.io/en/latest/
MIT License
2.59k stars 589 forks source link

[JAX] Meta Learning Tutorial? #125

Open ImahnShekhzadeh opened 1 year ago

ImahnShekhzadeh commented 1 year ago

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 commented 1 year ago

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 :)