MAINT: dense state matrix and array representation work