@yzh119 proposed this in #67. I'm pulling it into its own issue to track separately, so we can close that old large tracking issue.