Interpretable VAEs for nonlinear group factor analysis