Phillip Lippe

No followers

community-curated profile

PhD candidate at @UvA_Amsterdam (@quvalab), working on causal representation learning | ML @GoogleDevExpert for JAX+Flax | Intern @MSFTResearch

Overview Posts Content Recommendations

Article

Introduction to JAX+Flax

by Phillip Lippe

Recommended by 1 person

1 mention

Tweet Jun 13, 2022

Are you interested in learning JAX with Flax? We have translated our popular Deep Learning tutorials on CNNs, GNNs, (Vision) Transformers, and more from PyTorch to JAX+Flax, with considerable speedups for smaller models! Check them out here: uvadlc-n

by Phillip Lippe