* simplify local attention * update tests * add a comment and use torch.bitwise_xor
Attach a file by drag & drop or click to upload