@article{tensorprogramoptimizationwith,
  title = {Tensor Program Optimization with Probabilistic Programs},
  author = {Junru Shao and Xiyou Zhou and Siyuan Feng and Bohan Hou and Ruihang Lai and Hongyi Jin and Wuwei Lin and Masahiro Masuda and Cody Hao Yu and Tianqi Chen},
  year = {2022},
  eprint = {2205.13603},
  archivePrefix = {arXiv},
  url = {https://arxiv.org/abs/2205.13603v2},
}