CyberZHG / torch-multi-head-attentionLinks
Multi-head attention in PyTorch
☆154Updated 6 years ago
Alternatives and similar repositories for torch-multi-head-attention
Users that are interested in torch-multi-head-attention are comparing it to the libraries listed below
Sorting:
- Implement the paper "Self-Attention with Relative Position Representations"☆138Updated 4 years ago
- This in my Demo of Chen et al. "GradNorm: Gradient Normalization for Adaptive Loss Balancing in Deep Multitask Networks" ICML 2018☆179Updated 3 years ago
- Experiments with supervised contrastive learning methods with different loss functions☆221Updated 2 years ago
- PyTorch implementation of Representation Learning with Contrastive Predictive Coding by Van den Oord et al. (2018)☆87Updated 3 years ago
- Pytorch implementation of the GradNorm. GradNorm addresses the problem of balancing multiple losses for multi-task learning by learning a…☆270Updated 3 years ago
- Implementing SYNTHESIZER: Rethinking Self-Attention in Transformer Models using Pytorch☆70Updated 5 years ago
- Transformer/Transformer-XL/R-Transformer examples and explanations☆26Updated 3 years ago
- Code implementing the experiments described in the paper "On The Power of Curriculum Learning in Training Deep Networks" by Hacohen & Wei…☆114Updated 5 years ago
- A quick walk-through of the innards of LSTMs and a naive implementation of the Mogrifier LSTM paper in PyTorch☆78Updated 5 years ago
- MSc group project: Reproduction of 'Multi-Task Learning using Uncertainty to Weigh Losses for Scene Geometry and Semantics'; A. Kendall, …☆91Updated 5 years ago
- ☆83Updated 5 years ago
- an implementation of Deep Variational Informational Bottleneck in pytorch (https://arxiv.org/pdf/1612.00410.pdf)☆33Updated 7 years ago
- The Noise Contrastive Estimation for softmax output written in Pytorch☆319Updated 5 years ago
- [ICML 2021 Oral] We show pure attention suffers rank collapse, and how different mechanisms combat it.☆166Updated 4 years ago
- Pytorch implementation of R-Transformer. Some parts of the code are adapted from the implementation of TCN and Transformer.☆230Updated 6 years ago
- This repository contain various types of attention mechanism like Bahdanau , Soft attention , Additive Attention , Hierarchical Attention…☆125Updated 3 years ago
- Unofficial implementation of Google's FNet: Mixing Tokens with Fourier Transforms☆259Updated 4 years ago
- This is a repository for Multi-task learning with toy data in Pytorch and Tensorflow☆136Updated 6 years ago
- Code for "Understanding and Improving Layer Normalization"☆46Updated 5 years ago
- PCGrad pytorch sample code [not official]☆32Updated 5 years ago
- LSTM and GRU in PyTorch☆262Updated 6 years ago
- A Tensorflow implementation of the paper arXiv:1604.03539☆133Updated 7 years ago
- PyTorch implementation of Pay Attention to MLPs☆40Updated 4 years ago
- A pytorch &keras implementation and demo of Fastformer.☆189Updated 2 years ago
- pytorch neural network attention mechanism☆147Updated 6 years ago
- A PyTorch implementation of the Compact Multi-Head Self-Attention Mechanism from the paper: "Low Rank Factorization for Compact Multi-Hea…☆25Updated 5 years ago
- A PyTorch implementation of the paper - "Synthesizer: Rethinking Self-Attention in Transformer Models"☆73Updated 2 years ago
- Minimal RNN classifier with self-attention in Pytorch☆151Updated 3 years ago
- Learning to Encode Position for Transformer with Continuous Dynamical Model☆60Updated 5 years ago
- For paper《Gaussian Transformer: A Lightweight Approach for Natural Language Inference》☆28Updated 5 years ago