Deep Learning tutorials with JAX+Flax by Phillip Lippe
A series of notebooks explaining various deep learning concepts, from basics (e.g. intro to JAX/Flax, activiation functions) to recent advances (e.g., Vision Transformers, SimCLR), with translations to PyTorch.
- from
- JAX
- added
- 2026-10-10
- likes
- 0
similar
-
-
-
Tutorial: image classification with JAX and Flax Linen by 8bitmp3 github.com
Learn how to create a simple convolutional network with the Linen API by Flax and train it to recognize handwritten digits.
-
-
JAX, Flax & Transformers 🤗 github.com
3 days of talks around JAX / Flax, Transformers, large-scale language modeling and other great topics.
-
EasyLM github.com
LLMs made easy: Pre-training, finetuning, evaluating and serving LLMs in JAX/Flax.
JAX › Tutorials and Blog Posts: “A series of notebooks explaining various deep learning concepts, from basics (e.g. intro to JAX/Flax, activiation functions) to recent advances (e.g., Vision Transformers, SimCLR), with translations to PyTorch.”