You'll need more vertices on your plane to do the sort of interpolation you want to do. Upscale your vertex and texcoord matrix to like ~20x20 or something using jit.matrix @interp 1 and you should be set.

Best, Andrew B.