Use ndarray for quantiles in jax percentile

Joshuaalbert / jaxns

Probabilistic Programming and Nested sampling in JAX

https://jaxns.readthedocs.io/

Other

141 stars 9 forks source link

Closed fehiepsi closed 2 years ago

fehiepsi commented 2 years ago

Since jax 0.2.21, list argument is not supported in jnp.percentile. This PR fixes that issue.

Joshuaalbert commented 2 years ago

This will be address in Release 1.0, see #44.

fehiepsi commented 2 years ago

Thanks, I guess I can close this in that case.