Learned Transformers
https://ai.googleblog.com/2022/02/nested-hierarchical-transformer-towards.html?m=1