├── .github └── FUNDING.yml ├── LICENSE ├── README.md ├── images ├── ats.png ├── cait.png ├── cross_vit.png ├── crossformer.png ├── crossformer2.png ├── cvt.png ├── dino.png ├── distill.png ├── esvit.png ├── learnable-memory-vit.png ├── levit.png ├── mae.png ├── max-vit.png ├── mbvit.png ├── nest.png ├── parallel-vit.png ├── patch_merger.png ├── pit.png ├── regionvit.png ├── regionvit2.png ├── scalable-vit-1.png ├── scalable-vit-2.png ├── sep-vit.png ├── simmim.png ├── t2t.png ├── twins_svt.png ├── vit.gif └── vit_for_small_datasets.png └── vit_flax ├── cait.py ├── cross_vit.py ├── crossformer.py ├── cvt.py ├── deepvit.py ├── efficient.py ├── levit.py ├── nest.py ├── parallel_vit.py ├── patch_merger.py ├── pit.py ├── region_vit_flax.py ├── scalable_vit.py ├── simple_vit.py ├── t2t.py ├── twins_svt.py ├── vit.py └── vit_small_datasets.py /.github/FUNDING.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conceptofmind/vit-flax/HEAD/.github/FUNDING.yml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conceptofmind/vit-flax/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conceptofmind/vit-flax/HEAD/README.md -------------------------------------------------------------------------------- /images/ats.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conceptofmind/vit-flax/HEAD/images/ats.png -------------------------------------------------------------------------------- /images/cait.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conceptofmind/vit-flax/HEAD/images/cait.png -------------------------------------------------------------------------------- /images/cross_vit.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conceptofmind/vit-flax/HEAD/images/cross_vit.png -------------------------------------------------------------------------------- /images/crossformer.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conceptofmind/vit-flax/HEAD/images/crossformer.png -------------------------------------------------------------------------------- /images/crossformer2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conceptofmind/vit-flax/HEAD/images/crossformer2.png -------------------------------------------------------------------------------- /images/cvt.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conceptofmind/vit-flax/HEAD/images/cvt.png -------------------------------------------------------------------------------- /images/dino.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conceptofmind/vit-flax/HEAD/images/dino.png -------------------------------------------------------------------------------- /images/distill.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conceptofmind/vit-flax/HEAD/images/distill.png -------------------------------------------------------------------------------- /images/esvit.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conceptofmind/vit-flax/HEAD/images/esvit.png -------------------------------------------------------------------------------- /images/learnable-memory-vit.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conceptofmind/vit-flax/HEAD/images/learnable-memory-vit.png -------------------------------------------------------------------------------- /images/levit.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conceptofmind/vit-flax/HEAD/images/levit.png -------------------------------------------------------------------------------- /images/mae.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conceptofmind/vit-flax/HEAD/images/mae.png -------------------------------------------------------------------------------- /images/max-vit.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conceptofmind/vit-flax/HEAD/images/max-vit.png -------------------------------------------------------------------------------- /images/mbvit.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conceptofmind/vit-flax/HEAD/images/mbvit.png -------------------------------------------------------------------------------- /images/nest.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conceptofmind/vit-flax/HEAD/images/nest.png -------------------------------------------------------------------------------- /images/parallel-vit.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conceptofmind/vit-flax/HEAD/images/parallel-vit.png -------------------------------------------------------------------------------- /images/patch_merger.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conceptofmind/vit-flax/HEAD/images/patch_merger.png -------------------------------------------------------------------------------- /images/pit.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conceptofmind/vit-flax/HEAD/images/pit.png -------------------------------------------------------------------------------- /images/regionvit.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conceptofmind/vit-flax/HEAD/images/regionvit.png -------------------------------------------------------------------------------- /images/regionvit2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conceptofmind/vit-flax/HEAD/images/regionvit2.png -------------------------------------------------------------------------------- /images/scalable-vit-1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conceptofmind/vit-flax/HEAD/images/scalable-vit-1.png -------------------------------------------------------------------------------- /images/scalable-vit-2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conceptofmind/vit-flax/HEAD/images/scalable-vit-2.png -------------------------------------------------------------------------------- /images/sep-vit.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conceptofmind/vit-flax/HEAD/images/sep-vit.png -------------------------------------------------------------------------------- /images/simmim.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conceptofmind/vit-flax/HEAD/images/simmim.png -------------------------------------------------------------------------------- /images/t2t.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conceptofmind/vit-flax/HEAD/images/t2t.png -------------------------------------------------------------------------------- /images/twins_svt.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conceptofmind/vit-flax/HEAD/images/twins_svt.png -------------------------------------------------------------------------------- /images/vit.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conceptofmind/vit-flax/HEAD/images/vit.gif -------------------------------------------------------------------------------- /images/vit_for_small_datasets.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conceptofmind/vit-flax/HEAD/images/vit_for_small_datasets.png -------------------------------------------------------------------------------- /vit_flax/cait.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conceptofmind/vit-flax/HEAD/vit_flax/cait.py -------------------------------------------------------------------------------- /vit_flax/cross_vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conceptofmind/vit-flax/HEAD/vit_flax/cross_vit.py -------------------------------------------------------------------------------- /vit_flax/crossformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conceptofmind/vit-flax/HEAD/vit_flax/crossformer.py -------------------------------------------------------------------------------- /vit_flax/cvt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conceptofmind/vit-flax/HEAD/vit_flax/cvt.py -------------------------------------------------------------------------------- /vit_flax/deepvit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conceptofmind/vit-flax/HEAD/vit_flax/deepvit.py -------------------------------------------------------------------------------- /vit_flax/efficient.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conceptofmind/vit-flax/HEAD/vit_flax/efficient.py -------------------------------------------------------------------------------- /vit_flax/levit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conceptofmind/vit-flax/HEAD/vit_flax/levit.py -------------------------------------------------------------------------------- /vit_flax/nest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conceptofmind/vit-flax/HEAD/vit_flax/nest.py -------------------------------------------------------------------------------- /vit_flax/parallel_vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conceptofmind/vit-flax/HEAD/vit_flax/parallel_vit.py -------------------------------------------------------------------------------- /vit_flax/patch_merger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conceptofmind/vit-flax/HEAD/vit_flax/patch_merger.py -------------------------------------------------------------------------------- /vit_flax/pit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conceptofmind/vit-flax/HEAD/vit_flax/pit.py -------------------------------------------------------------------------------- /vit_flax/region_vit_flax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conceptofmind/vit-flax/HEAD/vit_flax/region_vit_flax.py -------------------------------------------------------------------------------- /vit_flax/scalable_vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conceptofmind/vit-flax/HEAD/vit_flax/scalable_vit.py -------------------------------------------------------------------------------- /vit_flax/simple_vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conceptofmind/vit-flax/HEAD/vit_flax/simple_vit.py -------------------------------------------------------------------------------- /vit_flax/t2t.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conceptofmind/vit-flax/HEAD/vit_flax/t2t.py -------------------------------------------------------------------------------- /vit_flax/twins_svt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conceptofmind/vit-flax/HEAD/vit_flax/twins_svt.py -------------------------------------------------------------------------------- /vit_flax/vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conceptofmind/vit-flax/HEAD/vit_flax/vit.py -------------------------------------------------------------------------------- /vit_flax/vit_small_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conceptofmind/vit-flax/HEAD/vit_flax/vit_small_datasets.py --------------------------------------------------------------------------------