CyberZHG / torch-multi-head-attention
Multi-head attention in PyTorch
☆152Updated 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
- Implement the paper "Self-Attention with Relative Position Representations"☆128Updated 4 years ago
- The Noise Contrastive Estimation for softmax output written in Pytorch☆318Updated 5 years ago
- Experiments with supervised contrastive learning methods with different loss functions☆220Updated 2 years ago
- Implementing SYNTHESIZER: Rethinking Self-Attention in Transformer Models using Pytorch☆70Updated 4 years ago
- ☆83Updated 5 years ago
- PyTorch implementation of Representation Learning with Contrastive Predictive Coding by Van den Oord et al. (2018)☆84Updated 3 years ago
- PyTorch implementation of Pay Attention to MLPs☆40Updated 3 years ago
- Contrastive Predictive Coding for Automatic Speaker Verification☆492Updated 5 years ago
- Pytorch implementation of R-Transformer. Some parts of the code are adapted from the implementation of TCN and Transformer.☆228Updated 5 years ago
- Unofficial implementation of Google's FNet: Mixing Tokens with Fourier Transforms☆259Updated 3 years ago
- MSc group project: Reproduction of 'Multi-Task Learning using Uncertainty to Weigh Losses for Scene Geometry and Semantics'; A. Kendall, …☆89Updated 5 years ago
- ☆64Updated 5 years ago
- Learning deep representations by mutual information estimation and maximization☆324Updated 6 years ago
- A pytorch &keras implementation and demo of Fastformer.☆188Updated 2 years ago
- a pytorch implementation of self-attention with relative position representations☆50Updated 4 years ago
- Independent implementation of Supervised Contrastive Loss. Straight to the point and beyond☆80Updated 4 years ago
- ☆33Updated 4 years ago
- Pytorch implementation of the GradNorm. GradNorm addresses the problem of balancing multiple losses for multi-task learning by learning a…☆269Updated 2 years ago
- Learning to Encode Position for Transformer with Continuous Dynamical Model☆59Updated 4 years ago
- Multihead Attention for PyTorch☆26Updated 6 years ago
- This in my Demo of Chen et al. "GradNorm: Gradient Normalization for Adaptive Loss Balancing in Deep Multitask Networks" ICML 2018☆178Updated 3 years ago
- [ICLR 2022] Official implementation of cosformer-attention in cosFormer: Rethinking Softmax in Attention☆190Updated 2 years ago
- Minimal RNN classifier with self-attention in Pytorch☆150Updated 3 years ago
- PCGrad pytorch sample code [not official]☆31Updated 4 years ago
- My take on a practical implementation of Linformer for Pytorch.☆415Updated 2 years ago
- code for Explicit Sparse Transformer☆61Updated last year
- pytorch implementation of Attention is all you need☆238Updated 3 years ago
- Reproducing the Linear Multihead Attention introduced in Linformer paper (Linformer: Self-Attention with Linear Complexity)☆76Updated 4 years ago
- Code implementing the experiments described in the paper "On The Power of Curriculum Learning in Training Deep Networks" by Hacohen & Wei…☆110Updated 5 years ago
- For paper《Gaussian Transformer: A Lightweight Approach for Natural Language Inference》☆28Updated 5 years ago