transformers
flax
git+https://github.com/patil-suraj/vqgan-jax.git