#
flax
Here are 2 public repositories matching this topic...
Jax/Flax implementation of DeiT and DeiT-III (ViT)
machine-learning
deep-learning
transformer
vision
image-classification
flax
tpu
jax
vision-transformer
timm
-
Updated
Apr 2, 2024 - Shell
Improve this page
Add a description, image, and links to the flax topic page so that developers can more easily learn about it.
Add this topic to your repo
To associate your repository with the flax topic, visit your repo's landing page and select "manage topics."