AIlibraryJupyter Notebook

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

25 stars2 forks2 issues更新于 2026/4/22

GitHub 仓库

暂无详细内容

同类推荐

AutoGPT

AutoGPT is the vision of accessible AI for everyone, to use and to build on. Our mission is to provide the tools, so that you can focus on what matters.

183,659

AutoGPT

AutoGPT is the vision of accessible AI for everyone, to use and to build on. Our mission is to provide the tools, so that you can focus on what matters.

183,659

transformers

🤗 Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal models, for both inference and training.

159,742

transformers

🤗 Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal models, for both inference and training.

159,742

whisper

Robust Speech Recognition via Large-Scale Weak Supervision

98,189

NextChat

87,764