tflahaul / jax-vision-transformers Public

Notifications You must be signed in to change notification settings
Fork 0
Star 4

Vision transformers with JAX & Flax (ViT, DeiT, LeViT, MAE, ConvPass)

4 stars 0 forks Branches Tags Activity

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 40 Commits
examples		examples
jvt		jvt
.gitignore		.gitignore
README.md		README.md
setup.py		setup.py

Repository files navigation

jax-vision-transformers

Vision transformers with JAX & Flax

Installation

pip install jvt

Usage example

from jvt import DeiT

ViT https://arxiv.org/abs/2010.11929
MAE https://arxiv.org/abs/2111.06377
DeiT https://arxiv.org/abs/2012.12877
LeViT https://arxiv.org/abs/2104.01136
ConvPass https://arxiv.org/abs/2207.07039