Retrieval-Enhanced Transformer – annotated PyTorch implementation
nn.labml.ai