@article{axeasimpleunifiedlayoutabstractionformac,
  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},
  eprint = {2601.19092},
  archivePrefix = {arXiv},
  url = {https://arxiv.org/abs/2601.19092},
}