Philippe Hansen-Estruch

icon indicating a website link https://philippe-eecs.github.io/website/

icon company UT Austin icon location Austin, TX icon location Interested in Generative Modeling, Reinforcement Learning, and Representation Learning

Results 4 comments of Philippe Hansen-Estruch

jax version

Hey, I've implemented a Jax version that I can release soon when I'm finished with my project :)!

optax.MultiSteps out of memory

I am having this issue as well for use in diffusion models

optax.MultiSteps out of memory

you're a king

jax version

Well "soon" is a relative term, but I just released it here! Includes an implementation of MAE and my own method. https://github.com/philippe-eecs/small-vision/tree/main