pyro-ppl / numpyro / 39302940725

93%

push

github

fix(distributions): account for the anchor in the simplex-to-ordered Jacobian (#2372)

* fix(distributions): account for the anchor in the simplex-to-ordered Jacobian

I found that nonzero anchor points changed the Jacobian used for simplex-to-ordered priors, giving incorrect transformed log probabilities. I subtract the anchor before evaluating the Jacobian and add scalar and batched regression coverage against a closed-form reference.

Assisted-by: Claude
Assisted-by: GPT-6 Astra
Signed-off-by: mottopanikeiku <176798723+mottopanikeiku@users.noreply.github.com>

* doc: drop the anchor note from SimplexToOrderedTransform

I removed the docstring sentence about the anchor, as suggested in review.

Assisted-by: Claude
Assisted-by: GPT-6 Astra
Signed-off-by: mottopanikeiku <176798723+mottopanikeiku@users.noreply.github.com>

---------

Signed-off-by: mottopanikeiku <176798723+mottopanikeiku@users.noreply.github.com>
Co-authored-by: mottopanikeiku <176798723+mottopanikeiku@users.noreply.github.com>

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

15753 of 16977 relevant lines covered (92.79%)

1.7 hits per line