Link to LSH class: https://github.com/leo4life2/algoml-final/blob/main/transformers/src/transformers/models/bert/modeling_lsh_bert.py#L72

Link to LSH attention: https://github.com/leo4life2/algoml-final/blob/main/transformers/src/transformers/models/bert/modeling_lsh_bert.py#L225