chem_vae_branch