changes for new alchemie + trainer for VAEs