Skip to content

Commit

Permalink
example of bug
Browse files Browse the repository at this point in the history
  • Loading branch information
@reubenharry
reubenharry committed Jun 2, 2024
1 parent f9652de commit 2669e4c
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion bug.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -103,7 +103,7 @@ def run_mclmc(logdensity_fn, num_steps, initial_position, transform, key, precon
key = jax.random.PRNGKey(0)


map = jax.pmap # change to pmap
map = jax.pmap # change to vmap

sampler = partial(run_mclmc, preconditioning=True)

Expand Down

0 comments on commit 2669e4c

Please sign in to comment.