Paper accepted in ICML. Collaboration with Eero Simocelliā€™s group led by Lyndon Duong and David Lipshutz. Congrats!