jarobyte91 / pytorch_beam_searchLinks
A lightweight implementation of Beam Search for sequence models in PyTorch.
☆55Updated 10 months ago
Alternatives and similar repositories for pytorch_beam_search
Users that are interested in pytorch_beam_search are comparing it to the libraries listed below
Sorting:
- ☆99Updated 3 years ago
- ☆176Updated 10 months ago
- pytorch; mask language model ; bert☆72Updated 5 years ago
- [TMLR'23] Contrastive Search Is What You Need For Neural Text Generation☆119Updated 2 years ago
- ☆72Updated 4 years ago
- Multilingual speech translation☆41Updated 4 years ago
- The original Backpack Language Model implementation, a fork of FlashAttention☆67Updated 2 years ago
- Systems submitted to IWSLT 2021 by the MT-UPC group.☆14Updated 2 years ago
- Reduce the size of pretrained Hugging Face models via vocabulary trimming.☆44Updated 2 years ago
- A PyTorch implementation of paper "Learning Shared Semantic Space for Speech-to-Text Translation", ACL (Findings) 2021☆48Updated 3 years ago
- Implementation of Imputer: Sequence Modelling via Imputation and Dynamic Programming in PyTorch☆58Updated 5 years ago
- Implementation of the GBST block from the Charformer paper, in Pytorch☆117Updated 3 years ago
- Train Dense Passage Retriever (DPR) with a single GPU☆131Updated 3 years ago
- ☆59Updated 2 years ago
- A toolkit for Spoken Language Understanding Evaluation (SLUE) benchmark. Refer paper https://arxiv.org/abs/2111.10367 for more details. O…☆65Updated last year
- Efficient Transformers with Dynamic Token Pooling☆61Updated 2 years ago
- 🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.☆82Updated 3 years ago
- ☆25Updated last year
- DiffusER: Discrete Diffusion via Edit-based Reconstruction (Reid, Hellendoorn & Neubig, 2022)☆54Updated 2 years ago
- This is the implementation of the paper AdaMix: Mixture-of-Adaptations for Parameter-efficient Model Tuning (https://arxiv.org/abs/2205.1…☆131Updated last year
- End-to-end Speech Translation with Stacked Acoustic-and-Textual Encoding☆26Updated 3 years ago
- PyTorch reimplementation of REALM and ORQA☆22Updated 3 years ago
- An official implementation of "BPE-Dropout: Simple and Effective Subword Regularization" algorithm.☆52Updated 4 years ago
- ☆43Updated last year
- Code for the NAACL 2022 long paper "DiffCSE: Difference-based Contrastive Learning for Sentence Embeddings"☆294Updated 2 years ago
- Tracking the progress in non-autoregressive generation (translation, transcription, etc.)☆306Updated 2 years ago
- Code and Models for the paper "End-to-End Training of Multi-Document Reader and Retriever for Open-Domain Question Answering" (NeurIPS 20…☆109Updated 3 years ago
- Using Pytorch's nn.Transformer module to create an english to french neural machine translation model.☆78Updated 4 years ago
- Sequence modeling with Mega.☆295Updated 2 years ago
- Official Implementation for the ICML2022 paper "Directed Acyclic Transformer for Non-Autoregressive Machine Translation"☆125Updated last year