neulab / knn-transformersLinks
PyTorch + HuggingFace code for RetoMaton: "Neuro-Symbolic Language Modeling with Automaton-augmented Retrieval" (ICML 2022), including an implementation of kNN-LM and kNN-MT
☆277Updated 2 years ago
Alternatives and similar repositories for knn-transformers
Users that are interested in knn-transformers are comparing it to the libraries listed below
Sorting:
- ☆322Updated 4 years ago
- Code for Editing Factual Knowledge in Language Models☆139Updated 3 years ago
- code associated with ACL 2021 DExperts paper☆115Updated 2 years ago
- ☆138Updated 7 months ago
- ☆184Updated last month
- ☆82Updated 2 years ago
- Automatic metrics for GEM tasks☆67Updated 2 years ago
- Tk-Instruct is a Transformer model that is tuned to solve many NLP tasks by following instructions.☆181Updated 2 years ago
- [EMNLP 2022] Training Language Models with Memory Augmentation https://arxiv.org/abs/2205.12674☆197Updated 2 years ago
- Train Dense Passage Retriever (DPR) with a single GPU☆133Updated 4 years ago
- An original implementation of "MetaICL Learning to Learn In Context" by Sewon Min, Mike Lewis, Luke Zettlemoyer and Hannaneh Hajishirzi☆270Updated 2 years ago
- A library for finding knowledge neurons in pretrained transformer models.☆158Updated 3 years ago
- The official code of EMNLP 2022, "SCROLLS: Standardized CompaRison Over Long Language Sequences".☆70Updated last year
- ☆87Updated 2 years ago
- ☆177Updated last year
- Code for paper "CrossFit : A Few-shot Learning Challenge for Cross-task Generalization in NLP" (https://arxiv.org/abs/2104.08835)☆112Updated 3 years ago
- ☆45Updated last year
- The LM Contamination Index is a manually created database of contamination evidences for LMs.☆78Updated last year
- The original implementation of Min et al. "Nonparametric Masked Language Modeling" (paper https//arxiv.org/abs/2212.01349)☆157Updated 2 years ago
- This repository accompanies our paper “Do Prompt-Based Models Really Understand the Meaning of Their Prompts?”☆85Updated 3 years ago
- ☆75Updated last year
- Package to compute Mauve, a similarity score between neural text and human text. Install with `pip install mauve-text`.☆295Updated last year
- ☆100Updated 3 years ago
- Code and models for the paper "Questions Are All You Need to Train a Dense Passage Retriever (TACL 2023)"☆63Updated 2 years ago
- Scalable training for dense retrieval models.☆299Updated 2 months ago
- [ICML 2023] Code for our paper “Compositional Exemplars for In-context Learning”.☆102Updated 2 years ago
- MEND: Fast Model Editing at Scale☆249Updated last year
- Repository for EMNLP 2022 Paper: Towards a Unified Multi-Dimensional Evaluator for Text Generation☆206Updated last year
- A unified benchmark for math reasoning☆88Updated 2 years ago
- [NeurIPS'22 Spotlight] Data and code for our paper CoNT: Contrastive Neural Text Generation☆154Updated 2 years ago