JAX-Flax-Tutorial-Image-Classification-with-Linen
How to use the Flax Linen API to build a convolutional neural network model and train it for image classification (using TensorFlow Datasets).
23 stars
2 watching
3 forks
Language: Jupyter Notebook
last commit: about 1 year ago
Linked from 1 awesome list
n2cholas/awesome-jax