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

BroadcastedDistribution #11

Open
mohamed82008 opened this issue Nov 1, 2019 · 1 comment
Open

BroadcastedDistribution #11

mohamed82008 opened this issue Nov 1, 2019 · 1 comment

Comments

@mohamed82008
Copy link
Member

Sometimes it is useful to have a broadcasted version of a distribution that is treated as a multivariate distribution. This is critical for performance when doing reverse AD using Tracker since we want to avoid scalar arithmetic. Defining an efficient adjoint for the BroadcastedDistribution will therefore lead to a significant improvement in speed.

@xukai92
Copy link
Member

xukai92 commented Jan 12, 2020

I think it's relatively straightforward to make BroadcastedDistribution GPU compatible as well.

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

No branches or pull requests

2 participants