lucidrains / memorizing-transformers-pytorch

Implementation of Memorizing Transformers (ICLR 2022), attention net augmented with indexing and retrieval of memories using approximate nearest neighbors, in Pytorch
MIT License
624 stars 46 forks source link

hugging face training code with demo #13

Open brando90 opened 1 year ago