Neural Discrete Representation Learning - arXiv