[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.