Yannick Wehr

LSH Reformer with multiple hashes not possible on TPU

1

### Description When attempting to train a Reformer with LSH Attention with n_hashes > 1 on a TPU, training will get stuck, and the trainer is not able to complete...