Resources
Here is a list of resources to get started with deep learning using JAX.
JAX
Other resources
- Awesome JAX list of resources
- JAX AI Stack tutorials
- Projects using the JAX AI Stack
Q&A
Data and pre-trained models
Data loaders
Data augmentation
Models and pre-trained weights
Flax
Official documentation
Other resources
Q&A
Alliance
The Alliance wiki has a page on Flax.