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