JAX-Flax-Tutorial-Image-Classification-with-Linen

JAX-Flax-Tutorial-Image-Classification-with-Linen

https://github.com/8bitmp3/JAX-Flax-Tutorial-Image-Classification-with-Linen

Jupyter Notebook21

8 months ago

Reviews

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

Search similar apps

    License

    Share

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

    Creator

    8bitmp3

    Related apps