Event Details

Tsunami: Training in TVM with Relay

Date: 12/16/2021 2:58 pm
Lightning Talks

Organization: OctoML
Speakers: Altan Haan
Download Slides

We present Tsunami, a work-in-progress training framework built on top of TVM using Relay. Tsunami lets us use TVM’s compilation flow to optimize and train BERT and DLRM. We also discuss some shortcomings that could be addressed by improving Relay.

Register for TVMCon 2021