-
Notifications
You must be signed in to change notification settings - Fork 146
Closed
Labels
Description
Description
We are defaulting to the numpyro implementation when that's installed, but the numpyro implementation is incredibly wasteful for large N, as it consists of doing n Categorical draws and summing up the values.
We should probably do sequential Binomial sampling: https://en.wikipedia.org/wiki/Multinomial_distribution#Algorithm:_Sequential_conditional_binomial_sampling