This embedding scheme allows us to represent each molecular fragment using a length d vector, where each entry takes value from {1, . . . , k} that corresponds to the dictionary embedding index for that row. The combinatorial vocabulary defined by the VQ-VAE has the capacity to represent d k distinct molecular fragments, which lifts the constraints of a limited generative span under a fixed fragment vocabulary.