LEARNING LATENT PERMUTATIONS WITH GUMBEL-SINKHORN NETWORKS PyTorch implementation of LEARNING LATENT PERMUTATIONS WITH GUMBEL-SINKHORN NETWORKS. Original implementation is available here. Requirement Pytorch torchvision numpy scipy Number Sorting cd number_sorting python train.py Jigsaw Puzzle cd jigsaw_puzzle python train.py