A Supplemental Material

A.1 Diagonal State Spaces We restate Proposition 1 for convenience. Equation 4 we have K " p Ce (Equation 7). This proves part (a) and we now consider part (b). (Equation 7). " 0 and hence softmax p0, iπq is not defined.