f0uriest / quadax Goto Github PK

Hi @f0uriest,

I've started using quadax in some of my projects and noticed that poetry complains about dependencies that could either be relaxed (numpy < 1.25.0 and, potentially, jax <= 0.4.14) or dropped entirely (scipy).

Thanks a lot for your work and help!

I use weighted quadrature in my research, and I was thinking of using quadax to replace scipy.integrate.quad in my work. I'd be interested in contributing a feature that handles weighted integration. Would you be open to a PR like that?

Hi, big fan of this repository! Are there any updates on n-dimensional quadrature? It would be very useful for me!

Hi @f0uriest,

Thanks for creating this neat package! For the example on the Readme, the assert statement is failing for me. The fix was to make JAX work with 64 bit precession with the lines

from jax import config
config.update("jax_enable_x64", True)

An alternative would be to remove the first assert statement and note that by default JAX using 32bit for floats. The second check works just fine but checking the precision out to 1e-14 might be too strict for other examples.

If you think this might be a bug, I'd be happy to contribute a PR for either of these fixes if you're interested.

Seems low priority, but could be nice to add cumulative_simpson integration. I think it should just involve copy-pasting https://docs.scipy.org/doc/scipy/reference/generated/scipy.integrate.cumulative_simpson.html and changing numpy np references to jax jnp.

Hi 👋 ,

Very cool library! I'm interested in using it for some numerical approximations in genetics, and was curious if its possible for users to specify points where the integral may have difficulties. This is similar to passing in points=... for scipy.integral.quad.