Diagonal State Spaces are as Effective as Structured State Spaces - Gupta, Ankit; Gu, Albert; Berant, Jonathan (Advances in Neural Information Processing Systems, 2022)