Add `scale_factor` to vae config. (#2)

Create README.md

flax

initial commit