Reparameterization_trick loading ...