andylolu2/jax-vqvae-gpt

Implementation of VQ-VAE with a GPT-style sampler in the JAX and Haiku ecosystem.

PythonMIT

Issues

Incorrect Codebook Loss Function Implementation in VQ-VAE
#3 opened 10 months ago by lars-quaedvlieg
1
Possible bug in VQ loss gradient
#2 opened 2 years ago by danijar
2