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).

updated at March 31, 2024, 6:02 p.m.

2 +0

22 +0

3 +0