Hi @jessegrabowski, thank you for your answer. That might be another problem I did not foresee, but I believe there’s still a problem with the indexing. For instance, if I remove “features” from coords and dims, and leave just the “time” dimension, I get the following:
If I instead just set F to 1, which should be essentially the same, I get degenerated posterior samples for Z:

