pyro-ppl / numpyro / 12978934612

92%

push

github

Add `log1mexp` and `logdiffexp` functions (#1960)

* Add log1mexp and logdiffexp functions and associated docs

* Move #noqa after a line break to avoid it getting included in link

* Clarify case of a == jnp.inf in logdiffexp docs

* Add needed extra line break

* Improve conditional structure, add basic tests

* Add type hints

* Add gradient tests

* Add more tests, including numerical tests of gradients

* Make custom jvp syntax follow jax docs, add some more tests

* Run ruff

* Run ruff

* Allow 'ans' in codespell, remove noqa statements

* Attribution for custom jvp approach

* Reformat files

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

13347 of 14376 relevant lines covered (92.84%)

1.97 hits per line