Skip to content

Commit 0152376

Browse files
authored
Remove vectorize_particles from RenyiELBO docstring (Issue #2166) (#2168)
* add mising parameters * undo
1 parent 878441d commit 0152376

1 file changed

Lines changed: 0 additions & 4 deletions

File tree

numpyro/infer/elbo.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -489,10 +489,6 @@ class RenyiELBO(ELBO):
489489
Here :math:`\alpha \neq 1`. Default is 0.
490490
:param num_particles: The number of particles/samples
491491
used to form the objective (gradient) estimator. Default is 2.
492-
:param vectorize_particles: Whether to use `jax.vmap` to compute ELBOs over the
493-
num_particles-many particles in parallel. If False use `jax.lax.map`.
494-
Defaults to True. You can also pass a callable to specify a custom vectorization
495-
strategy, for example `jax.pmap`.
496492
497493
Example::
498494

0 commit comments

Comments
 (0)