Skip to content

Commit

Permalink
Clean up Unrolled No U-Turn Sampler.
Browse files Browse the repository at this point in the history 
- Including trace_arrays in the previous_kernel_results to avoid mutating TransitionKernel state.
- Add divergence information to previous_kernel_results

PiperOrigin-RevId: 262445161
  • Loading branch information
@junpenglao @tensorflower-gardener
junpenglao authored and tensorflower-gardener committed Aug 8, 2019
1 parent 0986d2c commit 096aaa9
Showing 1 changed file with 198 additions and 185 deletions.
Loading

0 comments on commit 096aaa9

Please sign in to comment.