flax
feat: add configs for vae example
#3254
Merged

Loading