EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax
-
Updated
Jan 4, 2024 - Python
#
flax-linen
Here are 6 public repositories matching this topic...
GSoC'22 @ TensorFlow Notebooks, Code and much more
-
Updated
Sep 20, 2022 - Jupyter Notebook
A Simplistic trainer for Flax
-
Updated
Sep 22, 2024 - Python
Determine breed of dog in an image
machine-learning
kaggle
image-classification
vgg19
imagenet-dataset
dog-breed-classification
flax-linen
-
Updated
Jul 23, 2022 - Jupyter Notebook
A generalized implementation of Grad-CAM for Flax
-
Updated
Oct 23, 2023 - Python
Convert keras applications weights into flax model weights
-
Updated
Oct 14, 2023 - Python
Improve this page
Add a description, image, and links to the flax-linen topic page so that developers can more easily learn about it.
Add this topic to your repo
To associate your repository with the flax-linen topic, visit your repo's landing page and select "manage topics."