BobMcDear / vit-pytorchLinks
PyTorch implementation of the vision transformer
☆17Updated 2 years ago
Alternatives and similar repositories for vit-pytorch
Users that are interested in vit-pytorch are comparing it to the libraries listed below
Sorting:
- PyTorch implementation of EfficientNet☆10Updated 2 years ago
- Flax Image Models - State-of-the-art pre-trained vision backbones for Flax.☆19Updated 2 years ago
- Neural network from scratch in CUDA/C++☆80Updated 4 months ago
- PyTorch implementation of popular attention mechanisms in vision☆17Updated 2 years ago
- PyTorch implementation of SimSiam☆8Updated 2 years ago
- ☆16Updated 2 years ago
- Implementation of transformers based architecture in PyTorch.☆54Updated 4 years ago
- A set of of fundamental operations and deep learning models using JAX☆13Updated 4 years ago
- ☆51Updated 11 months ago
- CUDA implementation of autoregressive linear attention, with all the latest research findings☆44Updated 2 years ago
- Implementation of numerous Vision Transformers in Google's JAX and Flax.☆22Updated 2 years ago
- Implementation of the Kalman Filtering Attention proposed in "Kalman Filtering Attention for User Behavior Modeling in CTR Prediction"☆58Updated last year
- Skyformer: Remodel Self-Attention with Gaussian Kernel and Nystr\"om Method (NeurIPS 2021)☆61Updated 3 years ago
- Contains my experiments with the `big_vision` repo to train ViTs on ImageNet-1k.☆22Updated 2 years ago
- This repository hosts the code to port NumPy model weights of BiT-ResNets to TensorFlow SavedModel format.☆14Updated 3 years ago
- ☆39Updated 2 years ago
- Deep Learning Experiment Code.☆19Updated 10 months ago
- A multi-label text classifier to predict the subject areas of arXiv papers from their abstract bodies.☆17Updated 3 years ago
- Recursive Leasting Squares (RLS) with Neural Network for fast learning☆52Updated last year
- Local Attention - Flax module for Jax☆22Updated 4 years ago
- Implementation and explorations into Blackbox Gradient Sensing (BGS), an evolutionary strategies approach proposed in a Google Deepmind p…☆13Updated last week
- Examples of using PyTorch hooks, as covered in my YouTube tutorial video.☆33Updated last year
- FID computation in Jax/Flax.☆27Updated 10 months ago
- Implementation for ACProp ( Momentum centering and asynchronous update for adaptive gradient methdos, NeurIPS 2021)☆15Updated 3 years ago
- An implementation of the transformer architecture onto an Nvidia CUDA kernel☆183Updated last year
- Implementation of Token Shift GPT - An autoregressive model that solely relies on shifting the sequence space for mixing☆50Updated 3 years ago
- ☆23Updated 8 months ago
- ⛰️ RockyML - A High-Performance Scientific Computing Framework for Non-smooth Machine Learning Problems☆19Updated 2 years ago
- ☆53Updated 8 months ago
- Exploring an idea where one forgets about efficiency and carries out attention across each edge of the nodes (tokens)☆51Updated 2 months ago