> the basic NDArray backend shouldn't know anything about what number
> chain it is.

The NDArray could be rewritten to do without this, but I don't think
out-of-memory backends could do parallel sampling without knowing this
(and, even for NDArray, this is used to restart the same chain,
especially with parallel sampling).

Please let me know your thoughts. I'm happy to try different ideas and
rework this.