GitHub repository: rasamudano/vqgan-jax
Primary language: Python
Description: JAX implementation of VQGAN