Learning Latent Permutations with Gumbel-Sinkhorn Networks
TensorFlow: Large-Scale Machine Learning on Heterogeneous Distributed Systems