Fig. 6: Transformer architecture. | Nature Communications