Full paper accepted to Learning @ Scale 2021