/knn-transformers

PyTorch code for the RetoMaton paper: "Neuro-Symbolic Language Modeling with Automaton-augmented Retrieval" (ICML 2022), including an implementation of kNN-LM and kNN-MT

Primary LanguagePythonMIT LicenseMIT

No issues in this repository yet.