* add interpolate.py * Add jax dispatch for `searchsorted` * Import user-facing functions in `tensor.__init__`