ViT-jax2tf

This repository hosts code for converting the original Vision Transformer models (JAX) to TensorFlow.

APACHE-2.0 License

Stars
33
View Code on GitHub Visit Website View on X
Ecosystems: Tensorflow

Statistics for this project are still being loaded, please check back later.

Related Projects
MLPMixer-jax2tf

This repository hosts code for converting the original MLP Mixer models (JAX) to TensorFlow.

21 Sep 2021 13
deepvision

PyTorch and TensorFlow/Keras image models with automatic weight conversions and equal API/impleme...

31 Jan 2023 33
robustness-vit

Contains code for the paper "Vision Transformers are Robust Learners" (AAAI 2022).

12 Mar 2021 113
transformers

🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

29 Oct 2018 132,459
CvT-TensorFlow

TensorFlow 2.X reimplementation of CvT: Introducing Convolutions to Vision Transformers, Haiping ...

16 Oct 2022 3
swin-transformers-tf

Implementation of Swin Transformers in TensorFlow along with converted pre-trained models, code f...

08 May 2022 37
maxim-tf

Implementation of MAXIM in TensorFlow.

30 Sep 2022 116
probing-vits

Probing the representations of Vision Transformers.

12 Mar 2022 306
ConvNeXt-TF

Includes PyTorch -> Keras model porting code for ConvNeXt family of models with fine-tuning and i...

13 Jan 2022 92
cait-tf

Implementation of CaiT models in TensorFlow and ImageNet-1k checkpoints. Includes code for infere...

24 Apr 2022 9
deit-tf

Includes PyTorch -> Keras model porting code for DeiT models with fine-tuning and inference noteb...

27 Mar 2022 36
Transformer-in-Transformer

An Implementation of Transformer in Transformer in TensorFlow for image classification, attention...

03 Dec 2021 43
vision-transformers-tf

A non-exhaustive collection of vision transformer models implemented in TensorFlow.

25 Sep 2022 7
BiT-jax2tf

This repository hosts the code to port NumPy model weights of BiT-ResNets to TensorFlow SavedMode...

25 Aug 2021 13