https://github.com/affjljoo3581/deit3-jax

Jax/Flax implementation of DeiT and DeiT-III (ViT)
https://github.com/affjljoo3581/deit3-jax

deep-learning flax image-classification jax machine-learning timm tpu transformer vision vision-transformer

Last synced: 4 months ago
JSON representation

Jax/Flax implementation of DeiT and DeiT-III (ViT)

• Host: GitHub
• URL: https://github.com/affjljoo3581/deit3-jax
• Owner: affjljoo3581
• License: apache-2.0
• Created: 2024-03-27T00:50:26.000Z (over 2 years ago)
• Default Branch: main
• Last Pushed: 2024-12-21T08:52:40.000Z (over 1 year ago)
• Last Synced: 2025-03-25T06:34:11.794Z (over 1 year ago)
• Topics: deep-learning, flax, image-classification, jax, machine-learning, timm, tpu, transformer, vision, vision-transformer
• Language: Shell
• Homepage:
• Size: 53.7 KB
• Stars: 16
• Watchers: 2
• Forks: 6
• Open Issues: 0
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project