@misc{indiciae426de97dd1e0, title = {Axe: A Simple Unified Layout Abstraction for Machine Learning Compilers}, author = {Bohan Hou and Hongyi Jin and Guanjie Wang and Jinqi Chen and Yaxing Cai and Lijie Yang and Zihao Ye and Yaoyao Ding and Ruihang Lai and Tianqi Chen}, year = {2026}, url = {https://arxiv.org/abs/2601.19092}, note = {Source identifier: 2601.19092} }