LSH attention
Reformer uses LSH attention.