We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
vectorize_particles
RenyiELBO
1 parent 878441d commit 0152376Copy full SHA for 0152376
1 file changed
numpyro/infer/elbo.py
@@ -489,10 +489,6 @@ class RenyiELBO(ELBO):
489
Here :math:`\alpha \neq 1`. Default is 0.
490
:param num_particles: The number of particles/samples
491
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`.
496
497
Example::
498
0 commit comments