2d Galsim interpolants in JAX

[ismael-mendoza]

@beckermr implemented a jax version of 2D galsim interpolants in this notebook: https://github.com/beckermr/misc/blob/main/work/jax_dmdet_hacking/interp.ipynb

Might be good idea to integrate them into jax-galsim

[jecampagne]

Would be nice to get the residuals in a use-case for the 2D interpolants

[beckermr]

Oh yeah! It is on my list. I think though I'll just start a PR with tests to codify what they are explicitly.

[jecampagne]

FYI. Along the line to see if your code could be proposed to JAX, I was looking to Scipy 2D interpolation API, but (may be you know) the doc of scipy.interpolate.interp2d is mentioning that the API is deprecated since the 1.10.0 version and will be removed in the 1.13.0. The new API is documented here as scipy.interpolate.RegularGridInterpolator.

Now, if you are interested, I had opened a discussion in May 22 (!) on JAX here and I have proposed at that time code that you will find at the end of the thread. It would be nice to make some comparisons if you do not mind. My pb was the edges as usual...

[beckermr]

Feel free to compare away. My goal is to implement what is in galsim so that I can use it for other projects.