Implementation of Fast Weight Product Key Memories - View it on GitHub