Paper accepted in ICLR. Congratulations to David Lipshutz and Cengiz Pehlevan!