sayakpaul / ViT-jax2tfLinks
This repository hosts code for converting the original Vision Transformer models (JAX) to TensorFlow.
☆33Updated 3 years ago
Alternatives and similar repositories for ViT-jax2tf
Users that are interested in ViT-jax2tf are comparing it to the libraries listed below
Sorting:
- ☆24Updated 2 years ago
- This repository hosts the code to port NumPy model weights of BiT-ResNets to TensorFlow SavedModel format.☆14Updated 3 years ago
- Contains code to demonstrate distributed training in TensorFlow 2 with AI Platform and custom Docker contains.☆20Updated 4 years ago
- This repository hosts code for converting the original MLP Mixer models (JAX) to TensorFlow.☆15Updated 3 years ago
- Minimal implementation of PAWS (https://arxiv.org/abs/2104.13963) in TensorFlow.☆45Updated 4 years ago
- Implements MLP-Mixer (https://arxiv.org/abs/2105.01601) with the CIFAR-10 dataset.☆57Updated 3 years ago
- This project presents a simple framework to retrieve images similar to a query image.☆28Updated 4 years ago
- Contains Colab Notebooks show cool use-cases of different GCP ML APIs.☆10Updated 4 years ago
- ☆16Updated 2 years ago
- An attempt at the implementation of GLOM, Geoffrey Hinton's paper for emergent part-whole hierarchies from data☆37Updated 4 years ago
- Implementation of CaiT models in TensorFlow and ImageNet-1k checkpoints. Includes code for inference and fine-tuning.☆12Updated 2 years ago
- This repository provides a Colab Notebook that shows how to use Spatial Transformer Networks inside CNNs in Keras.☆37Updated 3 years ago
- Neural Arithmetic Logic Units by Trask et al.☆12Updated 6 years ago
- ☆15Updated 3 years ago
- Minimal implementation of Denoised Smoothing (https://arxiv.org/abs/2003.01908) in TensorFlow.☆20Updated 4 years ago
- Contains my experiments with the `big_vision` repo to train ViTs on ImageNet-1k.☆22Updated 2 years ago
- This project shows how to build a simple handwriting recognizer in Keras with the IAM dataset.☆13Updated 4 years ago
- Minimal implementation of adaptive gradient clipping (https://arxiv.org/abs/2102.06171) in TensorFlow 2.☆85Updated 4 years ago
- Implementation of modern data augmentation techniques in TensorFlow 2.x to be used in your training pipeline.☆34Updated 5 years ago
- TF2 implementation of knowledge distillation using the "function matching" hypothesis from https://arxiv.org/abs/2106.05237.☆87Updated 3 years ago
- Includes PyTorch -> Keras model porting code for DeiT models with fine-tuning and inference notebooks.☆41Updated 3 years ago
- Cyclemoid implementation for PyTorch☆90Updated 3 years ago
- Implements sharpness-aware minimization (https://arxiv.org/abs/2010.01412) in TensorFlow 2.☆60Updated 3 years ago
- Fastai community entry to 2020 Reproducibility Challenge☆17Updated 2 years ago
- Experiments with the ideas presented in https://arxiv.org/abs/2003.00152 by Frankle et al.☆29Updated 5 years ago
- A GPT, made only of MLPs, in Jax☆58Updated 4 years ago
- This project shows how to derive the total number of training tokens from a large text dataset from 🤗 datasets with Apache Beam and Data…☆27Updated 2 years ago
- ☆48Updated 4 years ago
- An implementation of Additive Attention☆149Updated 3 years ago
- An Easy To Use PyTorch Computer Vision Library☆52Updated 2 years ago