Right now when I call stack.apply(func, ...) the only choice I get is is volume — either it runs my…
Right now when I call stack.apply(func, ...) the only choice I get is is volume — either it runs my…: a task in MiMo-V2.6-RL-oss: Agentic RL Environments (MiMo RL release). Problem Statement Right now when I call stack.apply(func, ...) the only choice I get is is volume — either it runs my…
The task
Problem Statement Right now when I call stack.apply(func, ...) the only choice I get is is_volume — either it runs my function on 2D (Y,X) tiles or on the whole 3D (Z,Y,X) volume. But I've got cases where I want to process across other axes, like running something per-channel…
Part of XiaomiMiMo/MiMo-V2.6-RL-oss.