54k citations · 60.1k across the 24 of their papers we have counts for
Showing 2023Show all
2 papers · 1 filter
cs.LG2023★ 1 cited
ACRoBat: Optimizing Auto-batching of Dynamic Deep Learning at Compile Time
Pratik Fegade, Tianqi Chen, Phillip B. Gibbons +1
Dynamic control flow is an important technique often used to design expressive and efficient deep learning computations for applications such as text parsing, machine translation,…
cs.LG2023
ED-Batch: Efficient Automatic Batching of Dynamic Neural Networks via Learned Finite State Machines
Siyuan Chen, Pratik Fegade, Tianqi Chen +2
Batching has a fundamental influence on the efficiency of deep neural network (DNN) execution. However, for dynamic DNNs, efficient batching is particularly challenging as the data…