Building Generalizable Agents with a Realistic and Rich 3D Environment
Categorical Reparameterization with Gumbel-Softmax