https://towardsdatascience.com/hierarchical-transformers-part-2-2616eecacb21/