The mLSTMsig does not have a max state and therefore its state consists of a tuple of only two states (matC, vecN).
wrap_chunkwise__arbitrary_sequence_length() was written specifically for mLSTMexp kernels with a state consisting of a tuple of length 3 (matC, vecN, scaM).
Make wrap_chunkwise__arbitrary_sequence_length() handle states with arbitrary length or at least with length 2 and 3.