einsum fails when input arrays have different dtypes
einsum fails when input arrays have different dtypes: a task in MiMo-V2.6-RL-oss: Agentic RL Environments (MiMo RL release). einsum fails when input arrays have different dtypes I'm running into an issue when calling geomstats.backend.einsum with arrays of different dtypes. For example, mixing an…
The task
einsum fails when input arrays have different dtypes I'm running into an issue when calling geomstats.backend.einsum with arrays of different dtypes. For example, mixing an integer array with a float array (a common case when one of them is built from indices and the other from…
Part of XiaomiMiMo/MiMo-V2.6-RL-oss.