Product Key Memory

JSON →
library 0.3.0 ·python
verified May 9, 2026

A PyTorch implementation of Product Key Memory (PKM), an external memory module for neural networks with fast nearest-neighbor lookup via product quantization. Current version 0.3.0, requires Python >=3.6. Active development by lucidrains.

total hits 29
actors 5 distinct systems
last hit 19h ago AhrefsBot
Amazonbot
3
MetaBot
3
Search engines
15
Humans
2

top countries 🇺🇸 United States · 🇨🇦 Canada · 🇦🇺 Australia · 🇬🇧 United Kingdom · 🇳🇴 Norway