https://arashkhoeini.github.io/posts/FTTransformer/