knn-lm

There are 2 repositories under knn-lm topic.

  • neulab/knn-transformers

    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:Python26441123

  • neulab/retomaton

    PyTorch code for the RetoMaton paper: "Neuro-Symbolic Language Modeling with Automaton-augmented Retrieval" (ICML 2022)

    Language:Python67613