CyberZHG / torch-multi-head-attention
Multi-head attention in PyTorch
☆148Updated 5 years ago
Related projects ⓘ
Alternatives and complementary repositories for torch-multi-head-attention
- Implement the paper "Self-Attention with Relative Position Representations"☆124Updated 3 years ago
- Implementing SYNTHESIZER: Rethinking Self-Attention in Transformer Models using Pytorch☆70Updated 4 years ago
- Experiments with supervised contrastive learning methods with different loss functions☆216Updated last year
- PyTorch implementation of Representation Learning with Contrastive Predictive Coding by Van den Oord et al. (2018)☆82Updated 2 years ago
- pytorch implementation Variational Autoencoder and Conditional Variational Autoencoder☆77Updated 6 years ago
- ☆83Updated 5 years ago
- MSc group project: Reproduction of 'Multi-Task Learning using Uncertainty to Weigh Losses for Scene Geometry and Semantics'; A. Kendall, …☆84Updated 5 years ago
- A minimal pytorch package implementing a gradient reversal layer.☆156Updated 2 weeks ago
- The Noise Contrastive Estimation for softmax output written in Pytorch☆317Updated 5 years ago
- PCGrad pytorch sample code [not official]☆30Updated 4 years ago
- code for Explicit Sparse Transformer☆56Updated last year
- Reproducing the Linear Multihead Attention introduced in Linformer paper (Linformer: Self-Attention with Linear Complexity)☆73Updated 4 years ago
- A pytorch &keras implementation and demo of Fastformer.☆187Updated 2 years ago
- This is a repository for Multi-task learning with toy data in Pytorch and Tensorflow☆134Updated 6 years ago
- For paper《Gaussian Transformer: A Lightweight Approach for Natural Language Inference》☆27Updated 4 years ago
- Unofficial implementation of Google's FNet: Mixing Tokens with Fourier Transforms☆251Updated 3 years ago
- PyTorch implementation of Pay Attention to MLPs☆39Updated 3 years ago
- ☆64Updated 4 years ago
- Learning deep representations by mutual information estimation and maximization☆322Updated 5 years ago
- ☆32Updated 4 years ago
- lstm with layer normalization☆19Updated 3 years ago
- PyTorch implementation of "Effective Approaches to Attention-based Neural Machine Translation" using scheduled sampling to improve the pa…☆38Updated 7 years ago
- Pytorch implementation of the GradNorm. GradNorm addresses the problem of balancing multiple losses for multi-task learning by learning a…☆250Updated 2 years ago
- A collection of awesome things about mixed sample data augmentation☆132Updated 4 years ago
- [ICML 2021 Oral] We show pure attention suffers rank collapse, and how different mechanisms combat it.☆162Updated 3 years ago
- ☆144Updated 2 years ago
- Code for Multi-Head Attention: Collaborate Instead of Concatenate☆151Updated last year
- Code for "Understanding and Improving Layer Normalization"☆46Updated 4 years ago
- implement Domain-Adversarial Training of Neural Networks☆60Updated 6 years ago
- custom pytorch implementation of MoCo v3☆44Updated 3 years ago