Nystromformer for sequence length 512