direct optimization through arg max for discrete variational auto - encoder .
Attend, Infer, Repeat: Fast Scene Understanding with Generative Models