Allow multiple variable-rank dimension specs in check shapes
Allow multiple variable-rank dimension specs in check shapes: a task in MiMo-V2.6-RL-harbor-code: MiMo-V2.6-RL Code (Harbor) (Harbor dataset). This is hard for reasons similar to why GPflow/check shapes 5 is hard. In general, if there are multiple variable-rank dimensions in one specification, we…
The task
This is hard for reasons similar to why GPflow/check_shapes#5 is hard. In general, if there are multiple variable-rank dimensions in one specification, we cannot determine which dimensions belong to which variable. We could solve this is a couple of ways:
Part of FineEnvs/MiMo-V2.6-RL-harbor-code.