pyro-ppl / numpyro

92%

master: 92%

LAST BUILD ON BRANCH typing

branch: typing

CHANGE BRANCH

x

Reset

• typing
• InverseWishart
• 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
• 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-bernoulli-binomial-distributions
• doc-beta-dist
• doc-neg-binomial
• doc/gh-2187-chi2-discreteuniform-categorical
• doc/gh-2187-dirichlet-geometric
• doc/gh-2187-laplace-logistic-gumbel
• docs-2187-distribution-math
• docs/nuts-regularize-mass-matrix
• docs/rerun-gpjax-nb
• effect-handlers-tutorial
• elbo-sum-sites
• elbo-types
• ensemble-sampler-bf
• equinox-integration
• example-path
• expose-zeroinflated-dists
• external-samplers
• feat/haar-dct-time-reparam
• feat/schechter-distribution
• feature/arbitrary-initial-value
• fix-beta-edge-cases
• fix-deprecated-jnp-finfo-use
• 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-pr-triage
• fix-predictive-new-rng-keys
• fix-test
• fix-ty-config
• fix-xla-pmap-p-import
• fix/flax11-rng-dropout
• fix/jax-hoisted-provenance-constants
• 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
• 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
• 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
• 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
• scan-carry-shapes-list
• scope_divider_nn_module
• seed-block
• showfigure
• stateless_seed
• 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
• thomaspinder/gpjax-example
• thomaspinder/gpjax-simplified-api
• tombi-pre-commit
• truncated-cdf
• 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_extensions
• uncondition
• update-censored-nb
• update-toml-format
• ut-speed-cmp
• wishart
• wording-changes
• xfail-leak
• yaml-formatter
• zsh-compatibility

Build # 25250807924

Build Type

Pull #2178

github

Committed by web-flow

Commit Message

Merge 55fcbbcaa into 18c2cc135

Pull Request Pull Request #2178: Change key type from prng_key to Array

Coverage Stats

14107 of 15335 relevant lines covered (91.99%)

1.68 hits per line