Learning Robot Skill Embeddings
Categorical Reparameterization with Gumbel-Softmax