Refactor the ESS
computation and tests in blackjax.smc
#389
Labels
enhancement
New feature or request
refactoring
Change that adds no functionality but improves code quality
smc
Sequential Monte Carlo samplers
The current computation uses a
use_log
flag with branching logic that either applies the formula directly, or its numerical stable version. We can default to the numerically stable one, and usejax.scipy.logaddexp
instead of the custom logic.The tests for the ESS computation and the ESS solver appear to only test equality between the numerically stable version and the other one, and as such are not really testing for anything. Current tests should be replaced.
The text was updated successfully, but these errors were encountered: