See https://github.com/PaulLerner/ViQuAE