pyro-ppl / numpyro

93%

master: 93%

LAST BUILD ON BRANCH binomial-sampler-vmap-on-cpu

branch: binomial-sampler-vmap-on-cpu

CHANGE BRANCH

x

Reset

• binomial-sampler-vmap-on-cpu
• InverseWishart
• SimplexToOrderedTransform-inv
• a-an-usage
• add-jax-array-output-test
• add-log1mexp-logdiff-exp
• add-psis-diagnostic
• add-vsbc-diagnostic
• adjoint-leak
• agent/add-cat-constraint-transform
• arviz-autodims
• auto-review-workflow
• autoguide_signatures
• avoid-unnecessary-param-copies
• batched-scale-tril
• bayesian-var-tutorial
• bayesian_cuped_example
• betanegativebinomial
• better-type-hints-hsgp
• bin-x-ent-fix-nans
• bug/asvgd_memleak
• bugfix/mixture_distributions_no_rsample
• bugfix/mutables_leaves
• bugfix/nnx_module_mutables
• bugfix/null_mixture_weights_grad
• bump-to-0.16.1
• bump-to-0.17.0
• bump-to-0.18.0
• bump-to-0.19.0
• bump-to-0.20.0
• bump-to-0.21
• bump_release_0_20_1
• chore-format-README-with-ruff-0.16.0
• ci-pre-commit-hook
• ci-remove-continue-on-error
• circulant
• circular_gp_minor_docs_fixes
• codecov
• compatinility_nnx_12
• complex-biject-to
• composable-gibbs
• conf-py-typo
• consesus_mc
• correct-platform
• correct-typo
• cuda13
• dagum-dist
• delta-density
• delta-support
• deprecate-haiku
• develop
• dhm-fix-info-logdiffexp
• diagnostics-speedup
• dirichlet-multinom-total-count
• doc-2187-more
• doc-bernoulli-binomial-distributions
• doc-beta-dist
• doc-gh-2187-batch2
• doc-neg-binomial
• doc/2187-zeroinflatedpoisson-math
• doc/contrib-autoclass-reference-links
• doc/gaussian-state-space-math
• doc/gh-2079-contrib-cross-references
• doc/gh-2187-chi2-discreteuniform-categorical
• doc/gh-2187-dagum
• doc/gh-2187-delta
• doc/gh-2187-dirichlet-geometric
• doc/gh-2187-gaussian-random-walk
• doc/gh-2187-gompertz
• doc/gh-2187-laplace-logistic-gumbel
• doc/gh-2187-lkj-density
• doc/gh-2187-lognormal-loguniform
• doc/gh-2187-softlaplace
• doc/gh-2187-wishart
• doc/inverse-wishart-cholesky-math
• doc/inversegamma-kumaraswamy-gh2187
• doc/low-rank-mvn-docstring-gh-2187
• doc/ordered-logistic-docstring-gh-2187
• doc/relaxed-bernoulli-docstring-gh-2187
• doc/studentt-gh2187
• doc/vonmises-gh2187
• doc/wishart-cholesky-docs
• doc/zero-inflated-negative-binomial2-docstring-gh-2187
• docs-2187-distribution-math
• docs-car-distribution-issue-2187
• docs/2187-poisson-math-docs
• docs/add-claude-md
• docs/issue-2187-pareto-weibull
• docs/matrix-normal-details
• docs/numpyro-2187-halfnormal-uniform
• docs/nuts-regularize-mass-matrix
• docs/rerun-gpjax-nb
• docs/uv-contributing-setup
• docs/zero-sum-normal-covariance
• docstring-param-names
• effect-handlers-tutorial
• elbo-sum-sites
• elbo-types
• ensemble-sampler-bf
• enum-keyerror
• equinox-integration
• example-path
• expose-zeroinflated-dists
• external-samplers
• feat/condition-deterministic-rename
• feat/gaussian-hmm-family
• feat/gaussian-hmm-ops
• feat/haar-dct-time-reparam
• feat/schechter-distribution
• feature/arbitrary-initial-value
• fix-bernoulli-logits-entropy
• fix-beta-edge-cases
• fix-categorical-entropy-zero-probs
• fix-deprecated-jnp-finfo-use
• fix-discrete-support-offset
• fix-discrete-uniform-icdf
• fix-docs
• fix-docs-README
• fix-docs-build
• fix-entropy-samples-tests
• fix-error-handling
• fix-event-dim
• fix-fori-collect-recompilation
• fix-gaussian-state-space-variance-shape
• fix-get-model-relations-lambda
• fix-gpjax-docs
• fix-housing-dataset-path
• fix-issue-2013
• fix-issue-2123
• fix-jax-0.9.2
• fix-matrixnormal-doc
• fix-missing-param
• fix-optax-test
• fix-pmap-compilation-cache-leak
• fix-power-transform-log-det
• fix-pr-triage
• fix-predictive-new-rng-keys
• fix-test
• fix-ty-config
• fix-xla-pmap-p-import
• fix/cauchy-icdf-tails
• fix/ess-randommove-axis
• fix/extended-real-logits-constraints
• fix/flax11-rng-dropout
• fix/geometric-logits-moments
• fix/geometric-probs-deterministic-endpoint
• fix/gh-2299-intersphinx-inventories
• fix/gh-2301-ensemble-cached-log-density
• fix/jax-hoisted-provenance-constants
• fix/kumaraswamy-sample-cancellation
• fix/lognormal-small-scale-variance
• fix/loguniform-moment-ratio
• fix/negative-binomial-zero-mean
• fix/normal-entropy-scale-stability
• fix/pareto-entropy-logspace
• fix/weibull-small-cdf
• fix_missing_plus_sign
• fix_nb_version
• fix_uniform_supoprt
• flexible-optax-test
• fori-collect-eager-init
• format-with-ruff-0-15-0
• forward-diff-autoguide
• grammar-issues
• guideline
• hier-forecast
• hierarchical-feorecasting-example
• hot-fix-lint
• hsgp-rational-quadratic
• hurdle
• init-gaussian-state-space
• init-unconstrained
• intervalcensored
• issue-2126
• issue-2181
• jax-core
• jax-repo
• jax_82_depreciation
• key-reuse
• leak-dais
• leaky-plates
• leftrightcensored
• levy-dist
• master
• mc-stan-links
• mclmc
• mcmc-leak
• mcmc-progress-bar-diagnostics-sync
• mcmc-single-chain-states-copy
• mixture-logprob-unmasked-logsumexp
• mod1
• modify-README
• modify-type-error
• mutable-leak
• mypy-ty-migration
• nn-module-update-params-fix
• nnx-tracer-error
• nnx_integration
• nnx_merge_11_2
• no-plates
• optax_pass_value_arg
• ordabayevy-patch-1
• pass_rng_to_log_likelihood
• patch-1
• patch-2
• patch-3
• pin-ty
• poisson-zero-rate
• pr-cleanup
• pr-cleanup-update
• pred-mask
• prep-022-relase
• primitive-type-hints
• promote-batch
• pyproject.toml
• python311
• pytree
• pytree-support-fields
• reenable_haiku_tests
• refactor-prng-key
• replace-xxx-note
• repr
• resolve-pre-commit-warning
• rm-protocols
• safe-map
• scan-carry-shapes-list
• scope_divider_nn_module
• seed-block
• showfigure
• soft-vmap-skip-zero-pad
• stateless_seed
• stein-kernel-optional
• steinvi/rgn_kernel
• steinvi/simp-attr
• stochastic-example
• svi-progress-bar-host-sync
• template-config
• test-fix-for-jax-0-5-0
• test-gof
• test-python-3-13
• test-python3.14
• test/render-model-masked-regression
• thomaspinder/gpjax-example
• thomaspinder/gpjax-simplified-api
• tombi-pre-commit
• truncated-cdf
• truncated-gamma
• truncated-icdf
• truncated-variance
• try-fix-downaloads
• try-ty
• type-hint-constraints
• type-hint-distribution
• type-hint-transforms
• types-distributions
• types-handlers
• types-optim
• typing
• typing/distribution-sample-array
• typing_extensions
• uncondition
• update-censored-nb
• update-toml-format
• ut-speed-cmp
• wishart
• wording-changes
• xfail-leak
• yaml-formatter
• zsh-compatibility

Build # 37194974991

Build Type

Pull #2321

github

Committed by web-flow

Commit Message

Merge 74cd566bb into c17927606

Pull Request Pull Request #2321: perf(distributions): use vmap for the binomial sampler on CPU as well

Coverage Stats

16 of 16 new or added lines in 1 file covered. (100.0%)

23 existing lines in 1 file now uncovered.

15653 of 16876 relevant lines covered (92.75%)

1.7 hits per line