Awesome Lists

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

by 8bitmp3

archivedJupyter Notebookpushed about 3 years ago

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

stars
24
forks
3
watching
2
awesome list
1
View on GitHub

Embed the badge

Show how many awesome lists link to your project. The count updates automatically.

Awesome Lists badge
Markdown
[![Awesome Lists Badge](https://awesome.facts.dev/shield/8bitmp3/JAX-Flax-Tutorial-Image-Classification-with-Linen/links.svg)](https://awesome.facts.dev/awesome/8bitmp3/JAX-Flax-Tutorial-Image-Classification-with-Linen)
HTML
<a href="https://awesome.facts.dev/awesome/8bitmp3/JAX-Flax-Tutorial-Image-Classification-with-Linen"><img src="https://awesome.facts.dev/shield/8bitmp3/JAX-Flax-Tutorial-Image-Classification-with-Linen/links.svg" alt="Awesome Lists Badge" /></a>
Image URL
https://awesome.facts.dev/shield/8bitmp3/JAX-Flax-Tutorial-Image-Classification-with-Linen/links.svg

Add a GitHub project

Missing a project or an awesome list? Paste its GitHub URL and we fetch it right away.