Sinkhorn Transformer - Practical implementation of Sparse Sinkhorn Attention - View it on GitHub