direct optimization through arg max for discrete variational auto - encoder .
REBAR: Low-variance, unbiased gradient estimates for discrete latent variable models