A longer version of the ICLR paper exists on Arxiv that includes more experiments and theoretical guarantees (under review), this extended version is what is linked to here.