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-oss: Agentic RL Environments (MiMo RL release). I'm working with value-function iteration on a multi-dimensional state space built from a few 1-D grids (capital, productivity, etc.). The…
The task
I'm working with value-function iteration on a multi-dimensional state space built from a few 1-D grids (capital, productivity, etc.). The discretized state space is the cartesian product of these per-dimension grids, and I often need to map a continuous point back to the index…
Part of XiaomiMiMo/MiMo-V2.6-RL-oss.