Here we share our core transformer architecture implementation and the codebook updating module. 
We will be happy to share other parts of the code with reviewers upon request, including data loading and sampling, positional encoding generation, train and test functions, etc.