Basic bilinear interpolation on grids with jax. This package provides a class CartesianGrid
for data defined on a regular grid and interp2d(x, y, xp, yp, zp) for data on
irregular grids. CartesianGrid is a jax clone of the class from regulargrid.
Install by cloning the repository and running
pip install .You can install this directly via
python -m pip install git+https://github.com/adam-coogan/jaxinterp2d.git@masterThis will install jax and jaxlib if you don't already have them. The test
requires numpy and can be run with pytest. Check them out for a usage example
for interp2d.
Note that interp2d does not do any bounds checking.