Jake Vanderplas

Rendered docstrings preview here: https://jax--20799.org.readthedocs.build/en/20799/jax.scipy.html#module-jax.scipy.fft

Tracking issue: Documentation update

2

Started with #18269 The plan is to create an entirely new cohesive documentation structure in `docs/tutorials`, to move sections of the existing docs there, and then to remove the old...

BUG: scipy.stats.kstest returns NaN starting in scipy 1.12

8

### Describe your issue. When the input contains both negative and positive infinity, `scipy.stats.kstest` returns `NaN`. This appears to have been introduced in scipy 1.12.0 ### Reproducing Code Example ```python...

Code Organization

1

Currently, it's not clear from the package layout which routines are meant to be user-facing, and which routines are meant as utilities. Probably the easiest way to address this is...

We've had a couple requests recently for a way to define an assertion in a way that will execute normally outside JIT, but be ignored completely inside JIT. The thinking...

These are functions whose NumPy docstrings specifically call-out view semantics that don't apply in JAX.

[Possible bug] test suite requires value-dependent integer casting

1

The JAX array API I'm developing in https://github.com/google/jax/pull/16099 hits a number of failures due to pytest calling APIs with integers larger than the maximum `int64`. For example: ```python In [1]:...

Tests are too slow for libraries with eager-mode dispatch overhead

13

For example this job: https://github.com/google/jax/actions/runs/6804780902/job/18502996983, which is associated with https://github.com/google/jax/pull/16099 took 4.5 hours to run about 1000 tests. ``` ========== 567 failed, 439 passed, 116 skipped in 16419.47s (4:33:39) ==========...