Unofficial PyTorch implementation of Attention Free Transformer (AFT) layers by Apple Inc.
☆247Feb 16, 2026Updated 7 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
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- Code for the CVPR 2020 [ORAL] paper "SAM: The Sensitivity of Attribution Methods to Hyperparameters"☆29Dec 8, 2022Updated 3 years ago
- Multivariate Boosted TRee☆62Oct 3, 2022Updated 3 years ago
- Neural Ensemble Search for Uncertainty Estimation and Dataset Shift☆35Jan 10, 2026Updated 8 months ago
- ☆390Oct 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
- Drop-in replacement for any ResNet with a significantly reduced memory footprint and better representation capabilities☆206Apr 24, 2024Updated 2 years ago
- A GPT, made only of MLPs, in Jax☆59Jun 23, 2021Updated 5 years ago
- ☆42May 20, 2020Updated 6 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.
- 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
- 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 9 months ago
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- Implements the SM3-II adaptive optimization algorithm for PyTorch.☆33Sep 3, 2024Updated 2 years ago
- ☆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 2 months ago
- 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 2 months ago
- ☆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…☆431Aug 17, 2022Updated 4 years ago
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- Swarm training framework using Haiku + JAX + Ray for layer parallel transformer language models on unreliable, heterogeneous nodes☆241May 12, 2023Updated 3 years ago
- 👁️ Facial Landmark Annotation Tool with OpenCV☆12Apr 5, 2024Updated 2 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.☆145Jul 26, 2021Updated 5 years ago
- There and Back Again: Revisiting Backpropagation Saliency Methods (CVPR 2020)☆53Apr 7, 2020Updated 6 years ago