Shen Yan, Kaiqiang Song, Fei Liu, Mi Zhang: CATE: Computation-aware Neural Architecture Encoding with Transformers. ICML 2021: 11670-11681