TraceVAE