Tensor Program Optimization with Probabilistic Programs.

Sat, 01 Jan 2022 00:00:00 +0100

Junru Shao, Xiyou Zhou, Siyuan Feng, Bohan Hou, Ruihang Lai, Hongyi Jin, Wuwei Lin , Masahiro Masuda, Cody Hao Yu, Tianqi Chen:
Tensor Program Optimization with Probabilistic Programs. CoRR abs/2205.13603 (2022)