[WIP] JAX implementation of taming-transformers VQGAN.
(Note: For now this only consits of the
VQModel
for inference. The discriminatior, transformer model and training scripts will be added later.)
Use this to see how to use this model to encode and reconstruct images.