Skip to content

Commit

Permalink
For SNAPER, use the improved gradient estimator from Riou-Durand et al.
Browse files Browse the repository at this point in the history
The implementation required a bit of refactoring, since we need to be careful to only use validated (by MetropolisHastings) states for both previous and proposed states depending on direction.

Also, use harmonic mean for the docstring example, as that is best practice. The use of tfb.Exp() isn't ideal for tfd.HalfNormal(), but it gets to stay as it uncovered the issues mentioned above.

PiperOrigin-RevId: 607816872
  • Loading branch information
SiegeLordEx authored and tensorflower-gardener committed Feb 16, 2024
1 parent 045745e commit 0c26a85
Show file tree
Hide file tree
Showing 3 changed files with 187 additions and 92 deletions.
Loading

0 comments on commit 0c26a85

Please sign in to comment.