Baechi: Fast Device Placement of Machine Learning Graphs

Beomyeol Jeon, Linda Cai, Chirag Shetty +6

Machine Learning graphs (or models) can be challenging or impossible to train when either devices have limited memory, or models are large. To split the model across devices, learn…