This PR reintroduces parallel sampling from #446 , with upstream changes to tensor shape issues from https://github.com/BasisResearch/chirho/commit/49d1bf1e2d49a4de95acfdfc36b5a26482d587c6. This includes a minor change to the result processing, as the upstream tensor broadcasting adds a middle dimension to the tensor if the intervention has shape 1.
This PR reintroduces parallel sampling from #446 , with upstream changes to tensor shape issues from https://github.com/BasisResearch/chirho/commit/49d1bf1e2d49a4de95acfdfc36b5a26482d587c6. This includes a minor change to the result processing, as the upstream tensor broadcasting adds a middle dimension to the tensor if the intervention has shape 1.