A Continuous Relaxation of Beam Search for End-to-end Training of Neural Sequence Models
Categorical Reparameterization with Gumbel-Softmax