CyberZHG / torch-multi-head-attention
Multi-head attention in PyTorch
☆150Updated 5 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
- PyTorch implementation of Representation Learning with Contrastive Predictive Coding by Van den Oord et al. (2018)☆83Updated 3 years ago
- Experiments with supervised contrastive learning methods with different loss functions☆218Updated 2 years ago
- Implement the paper "Self-Attention with Relative Position Representations"☆125Updated 4 years ago
- ☆83Updated 5 years ago
- A quick walk-through of the innards of LSTMs and a naive implementation of the Mogrifier LSTM paper in PyTorch☆74Updated 4 years ago
- Pytorch implementation of the GradNorm. GradNorm addresses the problem of balancing multiple losses for multi-task learning by learning a…☆260Updated 2 years ago
- Implementing SYNTHESIZER: Rethinking Self-Attention in Transformer Models using Pytorch☆70Updated 4 years ago
- Multihead Attention for PyTorch☆25Updated 5 years ago
- Reproducing the Linear Multihead Attention introduced in Linformer paper (Linformer: Self-Attention with Linear Complexity)☆73Updated 4 years ago
- PyTorch implementation of Pay Attention to MLPs☆40Updated 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
- [ICML 2021 Oral] We show pure attention suffers rank collapse, and how different mechanisms combat it.☆164Updated 3 years ago
- pytorch implementation Variational Autoencoder and Conditional Variational Autoencoder☆78Updated 6 years ago
- Pytorch implementation of R-Transformer. Some parts of the code are adapted from the implementation of TCN and Transformer.☆227Updated 5 years ago
- The Noise Contrastive Estimation for softmax output written in Pytorch☆318Updated 5 years ago
- an implementation of Deep Variational Informational Bottleneck in pytorch (https://arxiv.org/pdf/1612.00410.pdf)☆33Updated 6 years ago
- About Code release for "Flowformer: Linearizing Transformers with Conservation Flows" (ICML 2022), https://arxiv.org/pdf/2202.06258.pdf☆311Updated 7 months ago
- This repository contain various types of attention mechanism like Bahdanau , Soft attention , Additive Attention , Hierarchical Attention…☆124Updated 3 years ago
- MSc group project: Reproduction of 'Multi-Task Learning using Uncertainty to Weigh Losses for Scene Geometry and Semantics'; A. Kendall, …☆87Updated 5 years ago
- A Pytorch Implementation of "Attention is All You Need" and "Weighted Transformer Network for Machine Translation"☆551Updated 4 years ago
- Code implementing the experiments described in the paper "On The Power of Curriculum Learning in Training Deep Networks" by Hacohen & Wei…☆108Updated 5 years ago
- ☆32Updated 4 years ago
- [ICLR 2022] Official implementation of cosformer-attention in cosFormer: Rethinking Softmax in Attention☆185Updated 2 years ago
- This in my Demo of Chen et al. "GradNorm: Gradient Normalization for Adaptive Loss Balancing in Deep Multitask Networks" ICML 2018☆172Updated 3 years ago
- code for Explicit Sparse Transformer☆60Updated last year
- Unofficial implementation of Google's FNet: Mixing Tokens with Fourier Transforms☆258Updated 3 years ago
- pytorch； mask language model ； bert☆72Updated 5 years ago
- This is the repository for "Efficient Low-rank Multimodal Fusion with Modality-Specific Factors", Liu and Shen, et. al. ACL 2018☆259Updated 4 years ago
- Learning deep representations by mutual information estimation and maximization☆324Updated 6 years ago
- Contrastive Predictive Coding for Automatic Speaker Verification☆486Updated 5 years ago