GitHub repository: lucidrains/memorizing-transformers-pytorch
Primary language: Python
Created: 2022-03-21T21:56:46Z
Description: Implementation of Memorizing Transformers (ICLR 2022), attention net augmented with indexing and retrieval of memories using approximate nearest neighbors, in Pytorch