We currently support a state transition using dist.Normal which has a shape (), whereas a state transition with dist.MVN has a shape (dim,). This forces us to juggle multiple shapes across the codebase. Instead, we should normalize this early on and have simpler expectations internally.
We currently support a state transition using
dist.Normalwhich has a shape(), whereas a state transition withdist.MVNhas a shape(dim,). This forces us to juggle multiple shapes across the codebase. Instead, we should normalize this early on and have simpler expectations internally.