Search Results for author: Pratik Fegade

Found 4 papers, 0 papers with code

ACRoBat: Optimizing Auto-batching of Dynamic Deep Learning at Compile Time

no code implementations17 May 2023 Pratik Fegade, Tianqi Chen, Phillip B. Gibbons, Todd C. Mowry

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, exiting early out of deep models and so on.

Code Generation Machine Translation +1

Cortex: A Compiler for Recursive Deep Learning Models

no code implementations2 Nov 2020 Pratik Fegade, Tianqi Chen, Phillip B. Gibbons, Todd C. Mowry

Optimizing deep learning models is generally performed in two steps: (i) high-level graph optimizations such as kernel fusion and (ii) low level kernel optimizations such as those found in vendor libraries.

Cannot find the paper you are looking for? You can Submit a new open access paper.