Standalone Product Key Memory module in Pytorch - for augmenting Transformer models
conda install conda-forge::product-key-memory