[ICML 2021 Oral] We show pure attention suffers rank collapse, and how different mechanisms combat it.
☆173Mar 8, 2021Updated 5 years ago
Alternatives and similar repositories for attention-rank-collapse
Users that are interested in attention-rank-collapse are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- ☆12Sep 26, 2019Updated 6 years ago
- ☆13Feb 16, 2021Updated 5 years ago
- Locally Enhanced Self-Attention: Rethinking Self-Attention as Local and Context Terms☆20Nov 29, 2021Updated 4 years ago
- Code for the TCS paper "On the performance of learned data structures" and the ICML paper "Why are learned indexes so effective?"☆21May 9, 2021Updated 5 years ago
- Code Release for the 2023 NeurIPS Paper How does GPT-2 compute greater-than?: Interpreting mathematical abilities in a pre-trained langua…☆17Dec 6, 2024Updated last year
- Deploy open-source AI quickly and easily - Special Bonus Offer • AdRunpod Hub is built for open source. One-click deployment and autoscaling endpoints without provisioning your own infrastructure.
- Code for Branched Schrödinger Bridge Matching (ICLR 2026)☆20Mar 20, 2026Updated 6 months ago
- Efficient Householder Transformation in PyTorch☆70Jul 6, 2021Updated 5 years ago
- [NeurIPS'20] Code for the Paper Compositional Visual Generation and Inference with Energy Based Models☆47Mar 24, 2023Updated 3 years ago
- ☆390Oct 18, 2023Updated 2 years ago
- Implementation of the GBST block from the Charformer paper, in Pytorch☆118Jul 15, 2021Updated 5 years ago
- A simple Transformer where the softmax has been replaced with normalization☆20Sep 11, 2020Updated 6 years ago
- Variational Walkback, NIPS'17☆28Oct 18, 2017Updated 8 years ago
- ☆42Mar 23, 2023Updated 3 years ago
- Implementation of RETRO, Deepmind's Retrieval based Attention net, in Pytorch☆878Oct 30, 2023Updated 2 years ago
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- ☆28Feb 26, 2026Updated 6 months ago
- ☆20Feb 26, 2021Updated 5 years ago
- Code for 'The Linear Representation Hypothesis and the Geometry of Large Language Models' (ICML 2024)☆132Feb 11, 2025Updated last year
- Official codebase for Pretrained Transformers as Universal Computation Engines.☆246Jan 14, 2022Updated 4 years ago
- Implementation of N-Grammer, augmenting Transformers with latent n-grams, in Pytorch☆82Dec 4, 2022Updated 3 years ago
- The source codes for D2AGE model. Distance-aware DAG Embedding for Proximity Search on Heterogeneous Graphs.☆12Feb 20, 2018Updated 8 years ago
- Training scripts for paper Miceli Barone et al. 2017 "Deep Architectures for Neural Machine Translation"☆11Jul 13, 2017Updated 9 years ago
- RelationNet++: Bridging Visual Representations for Object Detection via Transformer Decoder☆210Mar 18, 2021Updated 5 years ago
- MNIST, but with Bezier curves instead of pixels☆15Oct 29, 2021Updated 4 years ago
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- End-to-end training of sparse deep neural networks with little-to-no performance loss.☆338Jan 26, 2023Updated 3 years ago
- [CVPR 2021] Involution: Inverting the Inherence of Convolution for Visual Recognition, a brand new neural operator☆1,310Jul 16, 2021Updated 5 years ago
- ☆32Oct 13, 2021Updated 4 years ago
- ☆19Jun 21, 2025Updated last year
- Implementation of Fast Transformer in Pytorch☆176Aug 26, 2021Updated 5 years ago
- ☆45Mar 3, 2023Updated 3 years ago
- Fine-Tuning Pre-trained Transformers into Decaying Fast Weights☆20Oct 9, 2022Updated 3 years ago
- Implementation of ConvMixer for "Patches Are All You Need? 🤷"☆1,085Nov 11, 2022Updated 3 years ago
- [NeurIPS 2020]. COPT - Coordinated Optimal Transport on Graphs☆17Dec 22, 2020Updated 5 years ago
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- Disentangled Non-Local Neural Networks☆84Dec 7, 2020Updated 5 years ago
- computation of convolutional kernels (CKN and NTK) in C++☆14Dec 13, 2022Updated 3 years ago
- This is the official GitHub for paper: On the Versatile Uses of Partial Distance Correlation in Deep Learning, in ECCV 2022☆176Jun 4, 2023Updated 3 years ago
- Transformer based on a variant of attention that is linear complexity in respect to sequence length☆844May 5, 2024Updated 2 years ago
- Implementation of Lie Transformer, Equivariant Self-Attention, in Pytorch☆98Feb 19, 2021Updated 5 years ago
- Official repository for the "Big Transfer (BiT): General Visual Representation Learning" paper.☆1,541Jul 30, 2024Updated 2 years ago
- Code for 'Contrastive Multi-Document Question Generation'☆11Oct 16, 2022Updated 3 years ago