pyro-ppl / numpyro

93%

master: 92%

LAST BUILD ON BRANCH bugfix/nnx_module_mutables

branch: bugfix/nnx_module_mutables

CHANGE BRANCH

x

Reset

• bugfix/nnx_module_mutables
• add-log1mexp-logdiff-exp
• adjoint-leak
• arviz-autodims
• autoguide_signatures
• bayesian-var-tutorial
• bayesian_cuped_example
• bug/asvgd_memleak
• bump-to-0.16.1
• bump-to-0.17.0
• bump-to-0.18.0
• bump-to-0.19.0
• ci-remove-continue-on-error
• circulant
• circular_gp_minor_docs_fixes
• codecov
• compatinility_nnx_12
• complex-biject-to
• dagum-dist
• delta-density
• delta-support
• dhm-fix-info-logdiffexp
• dirichlet-multinom-total-count
• elbo-sum-sites
• elbo-types
• equinox-integration
• example-path
• feature/arbitrary-initial-value
• fix-entropy-samples-tests
• fix-event-dim
• fix-get-model-relations-lambda
• fix-matrixnormal-doc
• fix-predictive-new-rng-keys
• fix/flax11-rng-dropout
• fix_missing_plus_sign
• fix_nb_version
• guideline
• hierarchical-feorecasting-example
• init-unconstrained
• jax-core
• jax-repo
• key-reuse
• leak-dais
• leaky-plates
• levy-dist
• master
• mod1
• mutable-leak
• nn-module-update-params-fix
• nnx_integration
• nnx_merge_11_2
• no-plates
• optax_pass_value_arg
• pass_rng_to_log_likelihood
• patch-1
• primitive-type-hints
• promote-batch
• pytree
• reenable_haiku_tests
• repr
• safe-map
• scope_divider_nn_module
• seed-block
• showfigure
• stateless_seed
• steinvi/rgn_kernel
• steinvi/simp-attr
• template-config
• test-fix-for-jax-0-5-0
• test-gof
• test-python-3-13
• truncated-cdf
• truncated-icdf
• type-hint-distribution
• type-hint-transforms
• types-handlers
• types-optim
• typing_extensions
• wishart
• xfail-leak
• zsh-compatibility

Build # 16840614256

Build Type

Pull #2061

github

Committed by web-flow

Commit Message

Merge d9352f202 into d4411dc71

Pull Request Pull Request #2061: numpyro.contrib.module: only collect mutables in nnx_module() when th…

Run Details

13659 of 14709 relevant lines covered (92.86%)

1.98 hits per line