search

f0uriest / interpax

Interpolation and function approximation with JAX
MIT License
137 stars 11 forks source link

grad, jacfwd, jacrev #8

Open jecampagne opened 1 year ago

jecampagne commented 1 year ago

Nice job indeed. it would be nice to provide an example how to compute jax (grad, jacfwd, jacrev) on the interpolated function. Thanks

f0uriest commented 1 year ago

I just added some tests for this https://github.com/f0uriest/interpax/blob/e3c7455571265561dba40d4d4e3de3583def561b/tests/test_interpolate.py#L338

Depending on what you want to differentiate with respect to it will change a bit, let me know if it's not clear.