PyTorch + HuggingFace code for RetoMaton: "Neuro-Symbolic Language Modeling with Automaton-augmented Retrieval" (ICML 2022), including an implementation of kNN-LM and kNN-MT
language translation machine models transformers pytorch nearest-neighbor language-models knn neighbor nearest huggingface neuro-symbolic knn-lm retomaton knnlm knnmt knn-mt knn-transformers
-
Updated
Oct 20, 2022 - Python