JAX-Flax-Tutorial-Image-Classification-with-Linen
https://github.com/8bitmp3/JAX-Flax-Tutorial-Image-Classification-with-Linen
last year
Reviews
How to use the Flax Linen API to build a convolutional neural network model and train it for image classification (using TensorFlow Datasets).