Feature request: find nearest index in a cartesian grid without building the product

Feature request: find nearest index in a cartesian grid without building the product: a task in MiMo-V2.6-RL-harbor-code: MiMo-V2.6-RL Code (Harbor) (Harbor dataset). I'm working with value-function iteration on a multi-dimensional state space built from a few 1-D grids (capital, productivity…