Unofficial PyTorch implementation of Attention Free Transformer (AFT) layers by Apple Inc.
☆247Feb 16, 2026Updated 6 months ago
Alternatives and similar repositories for aft-pytorch
Users that are interested in aft-pytorch are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- GAN models implemented with Pytorch Lightning and Hydra configuration☆33Jun 5, 2022Updated 4 years ago
- Expressive Power of Invariant and Equivariant Graph Neural Networks (ICLR 2021)☆43Aug 25, 2023Updated 3 years ago
- Trains Transformer model variants. Data isn't shuffled between batches.☆145Oct 5, 2022Updated 3 years ago
- Official PyTorch Implementation of Long-Short Transformer (NeurIPS 2021).☆228Apr 18, 2022Updated 4 years ago
- Official code Cross-Covariance Image Transformer (XCiT)☆681Sep 28, 2021Updated 4 years ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- Code for the CVPR 2020 [ORAL] paper "SAM: The Sensitivity of Attribution Methods to Hyperparameters"☆29Dec 8, 2022Updated 3 years ago
- ☆12Jun 18, 2021Updated 5 years ago
- Multivariate Boosted TRee☆62Oct 3, 2022Updated 3 years ago
- Neural Ensemble Search for Uncertainty Estimation and Dataset Shift☆35Jan 10, 2026Updated 7 months ago
- ☆392Oct 18, 2023Updated 2 years ago
- bumble bee transformer☆14Apr 19, 2021Updated 5 years ago
- A collection of Models, Datasets, DataModules, Callbacks, Metrics, Losses and Loggers to better integrate pytorch-lightning with transfor…☆47May 29, 2023Updated 3 years ago
- Estimating Example Difficulty using Variance of Gradients☆67Jan 10, 2023Updated 3 years ago
- Drop-in replacement for any ResNet with a significantly reduced memory footprint and better representation capabilities☆206Apr 24, 2024Updated 2 years ago
- End-to-end encrypted cloud storage - Proton Drive • AdSpecial offer: 40% Off Yearly / 80% Off First Month. Protect your most important files, photos, and documents from prying eyes.
- A GPT, made only of MLPs, in Jax☆59Jun 23, 2021Updated 5 years ago
- ☆42May 20, 2020Updated 6 years ago
- http://nlp.seas.harvard.edu/2018/04/03/attention.html☆63May 20, 2021Updated 5 years ago
- [WIP] Behold, semantic-search, built over sentence-transformers to make it easy for search engineers to evaluate, optimise and deploy mod…☆15Apr 21, 2023Updated 3 years ago
- Implementation of Fast Transformer in Pytorch☆176Aug 26, 2021Updated 5 years ago
- Knowledge-Aware machine LEarning (KALE): accessible machine learning from multiple sources for interdisciplinary research, part of the 🔥…☆487Updated this week
- Code for "Self-Attention Between Datapoints: Going Beyond Individual Input-Output Pairs in Deep Learning"☆416Mar 21, 2024Updated 2 years ago
- Implementation of DropLoss for Long-Tail Instance Segmentation in Pytorch☆44Apr 14, 2021Updated 5 years ago
- Parallel waveform generation with DiffusionGAN☆17Mar 26, 2022Updated 4 years ago
- End-to-end encrypted cloud storage - Proton Drive • AdSpecial offer: 40% Off Yearly / 80% Off First Month. Protect your most important files, photos, and documents from prying eyes.
- Fast Block Sparse Matrices for Pytorch☆551Jan 21, 2021Updated 5 years ago
- Unofficial PyTorch implementation of the Involution layer from CVPR 2021☆45Dec 8, 2025Updated 8 months ago
- Implements the SM3-II adaptive optimization algorithm for PyTorch.☆33Sep 3, 2024Updated last year
- ☆100Dec 8, 2021Updated 4 years ago
- [ICML 2021] GraphNorm: A Principled Approach to Accelerating Graph Neural Network Training (official implementation)☆106Dec 19, 2022Updated 3 years ago
- Collection of machine learning research paper references☆27Jul 16, 2026Updated last month
- Official code for the paper "Why Do Self-Supervised Models Transfer? Investigating the Impact of Invariance on Downstream Tasks".☆16Dec 7, 2021Updated 4 years ago
- Implementation of Feedback Transformer in Pytorch☆108Mar 2, 2021Updated 5 years ago
- Simple dataset to dataloader library for pytorch☆32Jul 7, 2026Updated last month
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- ☆46Apr 13, 2022Updated 4 years ago
- An efficient implementation of the popular sequence models for text generation, summarization, and translation tasks. https://arxiv.org/p…☆433Aug 17, 2022Updated 4 years ago
- Swarm training framework using Haiku + JAX + Ray for layer parallel transformer language models on unreliable, heterogeneous nodes☆241May 12, 2023Updated 3 years ago
- Graph Data Augmentation Library for PyTorch Geometric☆134Aug 17, 2022Updated 4 years ago
- Implementation of Long-Short Transformer, combining local and global inductive biases for attention over long sequences, in Pytorch☆120Aug 4, 2021Updated 5 years ago
- Fully featured implementation of Routing Transformer☆300Nov 6, 2021Updated 4 years ago
- Code for the Shortformer model, from the ACL 2021 paper by Ofir Press, Noah A. Smith and Mike Lewis.☆147Jul 26, 2021Updated 5 years ago