Hello Otto.
In line 200 of kalmanfilter the following dimension check is made, along with others:
n_states = state_transition.shape[0]
Should this be
n_states = state_transition.shape[-2]
to support the use of multiple transition matrices, or is that not supported?
Peter