We note that<cite> Stern et al. (2019)</cite> also experimented with a number of other architectural variants, but we use the baseline version of the model described above in our experiments for simplicity.