Jamie Townsend

We were a bit careless in our implementation of the non-dyadic push and pop functions, and there's a bug, which is very rare at 64 bit precision. A reproduction, with...

Vmap of dynamic_update_slice very slow on TPU

5

### Description I've found that vmapping `lax.dynamic_update_slice` leads to surprising slow-downs. This issue affects the TPU backend. The following code benchmarks `vmap(dynamic_update_slice)` vs. an equivalent Python loop (plus `jnp.stack`): ```python...

MAINT: `stats.dirichlet_multinomial`: relax `n` to `>=0`

8

The methods in `scipy.stats.dirichlet_multinomial` (which was added [relatively recently](https://github.com/scipy/scipy/pull/17211)) call the function [`_dirichlet_multinomial_check_parameters`](https://github.com/scipy/scipy/blob/92d2a8592782ee19a1161d0bf3fc2241ba78bb63/scipy/stats/_multivariate.py#L5988-L6018) to validate their arguments. There is a requirement there that `n > 0`, which is stricter than...

Closes https://github.com/scipy/scipy/issues/21531, see that issue for the motivation of this change. Reference: https://en.wikipedia.org/wiki/Dirichlet-multinomial_distribution. Note that the article previously had n > 0, and I edited it on September 17th 2024....