Add chain awareness to shard identification
Add chain awareness to shard identification: a task in MiMo-V2.6-RL-oss: Agentic RL Environments (MiMo RL release). Right now a shard is identified by a 32-bit branch value that only encodes a shard size and a shard id. We are moving to a multi-chain layout where every shard also belongs to a…
The task
Right now a shard is identified by a 32-bit branch value that only encodes a shard size and a shard id. We are moving to a multi-chain layout where every shard also belongs to a chain, so the identifier needs to carry a chain id as well. The encoding Define a full shard id as a…
Part of XiaomiMiMo/MiMo-V2.6-RL-oss.