8bitmp3 / 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).
24Updated last year

Related projects

Alternatives and complementary repositories for JAX-Flax-Tutorial-Image-Classification-with-Linen