GitHub repository: tux-type/simple-vaekl
Primary language: Python
Description: Simple VAE-KL in Flax NNX.