Tutorial: image classification with JAX and Flax Linen by 8bitmp3
Jupyter Notebook
A tutorial on building and training a convolutional neural network for MNIST image classification using Flax Linen and Optax.
25 stars2 forks0 contributors
Overview
How to use the Flax Linen API to build a convolutional neural network model and train it for image classification (using TensorFlow Datasets).
Related Projects
Community-curated · Updated weekly · 100% open source
Found a gem we're missing?
Open-Awesome is built by the community, for the community. Submit a project, suggest an awesome list, or help improve the catalog on GitHub.
