Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Tutorial / Model Demonstrating limitations of scan with numpyro.plate #444

Open
damonbayer opened this issue Sep 12, 2024 · 1 comment
Open
Assignees
Labels

Comments

@damonbayer
Copy link
Collaborator

Our RandomVariables which use scan cannot be used in a numpyro.plate context (see https://num.pyro.ai/en/stable/primitives.html#scan). However, we can still write reasonable looking models by using numpyro.plate for non-scanning RandomVariables and inputting the results into our vectorized scanning RandomVariables. We should have a tutorial or model to demonstrate this limitation and our workaround.

@dylanhmorris
Copy link
Collaborator

I think for deterministic scans we should probably stick with jax.lax.scan but let's discuss in #444

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Projects
None yet
Development

No branches or pull requests

3 participants