JAX-Flax-Tutorial-Image-Classification-with-Linen in n2cholas/awesome-jax

How to use the Flax Linen API to build a convolutional neural network model and train it for image classification (using TensorFlow Datasets).

created at Dec. 24, 2020, 3 a.m.

2 +0

22 +0

3 +0