jax-ml / jax

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
http://jax.readthedocs.io/
Apache License 2.0
30.5k stars 2.8k forks source link

⚠️ Nightly upstream-dev CI failed ⚠️ #24875

Open github-actions[bot] opened 3 days ago

github-actions[bot] commented 3 days ago

Workflow Run URL

Summary of Failures ``` tests/lax_scipy_special_functions_test.py::LaxScipySpcialFunctionsTest::testGammaSign: AssertionError: Not equal to tolerance rtol=1e-05, atol=1e-15 nan location mismatch: ACTUAL: array([ nan, nan, nan, nan, nan, nan, nan, nan, nan, nan, inf, 1.000e+00, 1.000e+00, 2.000e+00, 6.000e+00, 2.400e+01, 1.200e+02, 7.200e+02, 5.040e+03, 4.032e+04]) DESIRED: array([ inf, inf, inf, inf, inf, inf, inf, inf, inf, inf, inf, 1.000e+00, 1.000e+00, 2.000e+00, 6.000e+00, 2.400e+01, 1.200e+02, 7.200e+02, 5.040e+03, 4.032e+04]) ```
jakevdp commented 2 days ago

Failing due to an upstream change in numerics: https://github.com/scipy/scipy/pull/21827

We should change the JAX implementation to match this.