pyro-ppl / numpyro

92%

master: 92%

LAST BUILD ON BRANCH fix-pmap-compilation-cache-leak

branch: fix-pmap-compilation-cache-leak

CHANGE BRANCH

x

Reset

• fix-pmap-compilation-cache-leak
• InverseWishart
• a-an-usage
• add-jax-array-output-test
• add-log1mexp-logdiff-exp
• add-psis-diagnostic
• add-vsbc-diagnostic
• adjoint-leak
• arviz-autodims
• autoguide_signatures
• bayesian-var-tutorial
• bayesian_cuped_example
• betanegativebinomial
• bug/asvgd_memleak
• bugfix/mixture_distributions_no_rsample
• bugfix/mutables_leaves
• bugfix/nnx_module_mutables
• 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
• ci-pre-commit-hook
• ci-remove-continue-on-error
• circulant
• circular_gp_minor_docs_fixes
• codecov
• compatinility_nnx_12
• complex-biject-to
• conf-py-typo
• consesus_mc
• correct-platform
• correct-typo
• cuda13
• dagum-dist
• delta-density
• delta-support
• deprecate-haiku
• dhm-fix-info-logdiffexp
• dirichlet-multinom-total-count
• doc-bernoulli-binomial-distributions
• doc-beta-dist
• doc-neg-binomial
• doc/gh-2187-chi2-discreteuniform-categorical
• doc/gh-2187-laplace-logistic-gumbel
• docs-2187-distribution-math
• docs/rerun-gpjax-nb
• effect-handlers-tutorial
• elbo-sum-sites
• elbo-types
• ensemble-sampler-bf
• equinox-integration
• example-path
• expose-zeroinflated-dists
• external-samplers
• feature/arbitrary-initial-value
• fix-beta-edge-cases
• fix-deprecated-jnp-finfo-use
• fix-docs
• 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-predictive-new-rng-keys
• fix-test
• fix-ty-config
• fix-xla-pmap-p-import
• fix/flax11-rng-dropout
• fix_missing_plus_sign
• fix_nb_version
• fix_uniform_supoprt
• flexible-optax-test
• format-with-ruff-0-15-0
• forward-diff-autoguide
• grammar-issues
• guideline
• hier-forecast
• hierarchical-feorecasting-example
• 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
• mod1
• 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
• poisson-zero-rate
• pred-mask
• primitive-type-hints
• promote-batch
• pyproject.toml
• python311
• pytree
• reenable_haiku_tests
• refactor-prng-key
• replace-xxx-note
• repr
• resolve-pre-commit-warning
• rm-protocols
• safe-map
• scope_divider_nn_module
• seed-block
• showfigure
• stateless_seed
• steinvi/rgn_kernel
• steinvi/simp-attr
• stochastic-example
• template-config
• test-fix-for-jax-0-5-0
• test-gof
• test-python-3-13
• test-python3.14
• thomaspinder/gpjax-example
• thomaspinder/gpjax-simplified-api
• tombi-pre-commit
• truncated-cdf
• truncated-icdf
• try-fix-downaloads
• try-ty
• type-hint-constraints
• type-hint-distribution
• type-hint-transforms
• types-distributions
• types-handlers
• types-optim
• typing
• typing_extensions
• uncondition
• update-toml-format
• wishart
• wording-changes
• xfail-leak
• yaml-formatter
• zsh-compatibility

Build # 24447615173

Build Type

Pull #2172

github

Committed by web-flow

Commit Message

Merge d36471078 into 0152376a1

Pull Request Pull Request #2172: mcmc: Fix memory leak when chain_method="parallel"

Coverage Stats

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

14110 of 15336 relevant lines covered (92.01%)

1.68 hits per line