Corpus ID: 119120115

Relay: A High-Level IR for Deep Learning

@article{Roesch2019RelayAH,
  title={Relay: A High-Level IR for Deep Learning},
  author={Jared Roesch and Steven Lyubomirsky and Marisa Kirisame and J. Pollock and Logan Weber and Ziheng Jiang and T. Chen and T. Moreau and Zachary Tatlock},
  journal={ArXiv},
  year={2019},
  volume={abs/1904.08368}
}

• Jared Roesch, Steven Lyubomirsky, +6 authors Zachary Tatlock
• Published 2019
• Mathematics, Computer Science
• ArXiv

Frameworks for writing, compiling, and optimizing deep learning (DL) models have recently enabled progress in areas like computer vision and natural language processing. Extending these frameworks to accommodate the rapidly diversifying landscape of DL models and hardware platforms presents challenging tradeoffs between expressiveness, composability, and portability. We present Relay, a new intermediate representation (IR) and compiler framework for DL models. The functional, statically-typed… Expand

View PDF on arXiv

Save to Library

Create Alert

Cite

Launch Research Feed