2
92%
master: 92%

Ran 28 Mar 2025 03:16AM UTC

Files 87

Run time 4s

Badge

Embed ▾

Committed 28 Mar 2025 02:38AM UTC coverage: 69.356% (-0.1%) from 69.503%

Job # 14120810158.2

Build Type

push

github

Committed by

web-flow

Commit Message

Equinox Integration (#2005)

* added initial support for equinox models

* removed accidental line spaces

* used alias for jax.tree_util in contrib/module

* added docstrings for update_tree_params

* working eqx_module that manages state internally

* fixed issue with state getting broadcasted

* added equinox as dev dependency

* made eqx_module functional

* better docstrings for equinox modules

* changed eqx smoke test to use autodelta

* stopped gradient for eqx state

* updated docstring to reflect stopping eqx state gradient

* removed update_tree_params and switched random_eqx_module to use _update_params

* added docs for eqx_modules, mutable. stopped gradient on mutable_state in svi loss_fn

* fixed docs

* adjusted test to not have latent x in model

* removed stopping state gradient in svi loss_fn

* updated docstring to reflect no gradient stopping

---------

Co-authored-by: kylejcaron <kylejcaron@bitbucket.org>

Coverage Stats

10171 of 14665 relevant lines covered (69.36%)

0.69 hits per line

pyro-ppl / numpyro / 14120810158 / 2
92%
master: 92%

README BADGES
x

Markdown

Textile

RDoc

HTML

Rst

Source Files on job test-modeling - 14120810158.2

pyro-ppl / numpyro / 14120810158 / 2 92% master: 92%

README BADGES x

Markdown

Textile

RDoc

HTML

Rst

Source Files on job test-modeling - 14120810158.2

pyro-ppl / numpyro / 14120810158 / 2
92%
master: 92%

README BADGES
x