pyro-ppl / numpyro

92%

LAST BUILD ON BRANCH master

branch: master

CHANGE BRANCH

x

Reset

• master
• InverseWishart
• a-an-usage
• add-log1mexp-logdiff-exp
• add-psis-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_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-neg-binomial
• elbo-sum-sites
• elbo-types
• equinox-integration
• example-path
• 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-get-model-relations-lambda
• fix-gpjax-docs
• fix-issue-2013
• fix-issue-2123
• fix-jax-0.9.2
• fix-matrixnormal-doc
• fix-optax-test
• fix-predictive-new-rng-keys
• fix-test
• fix-ty-config
• 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
• init-gaussian-state-space
• init-unconstrained
• intervalcensored
• issue-2126
• jax-core
• jax-repo
• jax_82_depreciation
• key-reuse
• leak-dais
• leaky-plates
• leftrightcensored
• levy-dist
• mc-stan-links
• mclmc
• mcmc-leak
• mod1
• mutable-leak
• nn-module-update-params-fix
• nnx-tracer-error
• nnx_integration
• nnx_merge_11_2
• no-plates
• optax_pass_value_arg
• pass_rng_to_log_likelihood
• patch-1
• pred-mask
• primitive-type-hints
• promote-batch
• pyproject.toml
• python311
• pytree
• reenable_haiku_tests
• refactor-prng-key
• 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
• 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_extensions
• uncondition
• wishart
• wording-changes
• xfail-leak
• yaml-formatter
• zsh-compatibility

Build # 23742377234

Build Type

push

github

Committed by web-flow

Commit Message

Fix: Tracer leaks during NUTS sampling on random_nnx_module when NNX module contains mutable state (#2162)

* fix

* rm skip

Run Details

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

14097 of 15323 relevant lines covered (92.0%)

1.68 hits per line