nystrom
Vision Xformers
An implementation of the Nyströmformer, using Nystrom method to approximate standard self attention
Monte-Carlo Nystrom method in NumPy / PyTorch