Skip to content

Instantly share code, notes, and snippets.

@Stella-S-Yan

Stella-S-Yan/jax_rq_vae.ipynb

Last active March 30, 2026 21:20
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save Stella-S-Yan/03830a1610dd64faf163b34ee01ad5fb to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save Stella-S-Yan/03830a1610dd64faf163b34ee01ad5fb to your computer and use it in GitHub Desktop.
Download ZIP
JAX Implementation of Residual-Quantized Variational Autoencoder (RQ-VAE)
Display the source blob
Display the rendered blob
Raw
jax_rq_vae.ipynb
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment