Su Yan (Stella) Stella-S-Yan
Stella-S-Yan
/ jax_rq_vae.ipynb
Last active
March 30, 2026 21:20
JAX Implementation of Residual-Quantized Variational Autoencoder (RQ-VAE)
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Stella-S-Yan
/ jax_transposed_conv_investigation.ipynb
Last active
March 6, 2025 23:18
Evaluate JAX conv & transposed_conv implementation
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.