Review [JAX] Fix jax.scipy.special.loggamma nan at zero (jax-ml/jax#38240) (#41864)

copybara-service[bot] authored commits starting at October 22

Diff Delta:

Classified as: Bug Fix

copybara-service[bot]'s Description of Work

[JAX] Fix jax.scipy.special.loggamma nan at zero (jax-ml/jax#38240)

Problem:
In jax-ml/jax#38240 (https://github.com/jax-ml/jax/issues/38240), jax.scipy.special.loggamma incorrectly returns nan at x=0.0 and x=-0.0 due to a strict x > 0 filter, whereas SciPy returns inf.

Solution:
Changed the filter in loggamma to use x >= 0 instead of x > 0 to correctly return inf at 0.0 and -0.0. Added a specific boundary value test in lax_scipy_special_functions_test.py.

Fixes https://github.com/jax-ml/jax/issues/38240

2 total changed files

Loading changes...

You’ve reached the end

Thus concludes today’s episode of “Just One Small Change”

[JAX] Fix jax.scipy.special.loggamma nan at zero (jax-ml/jax#38240) #41864

copybara-service[bot]'s Description of Work