einsum fails when input arrays have different dtypes

einsum fails when input arrays have different dtypes: a task in MiMo-V2.6-RL-harbor-code: MiMo-V2.6-RL Code (Harbor) (Harbor dataset). 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…

The task

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 coordinates) blows up the call instead of just doing the contraction.

Part of FineEnvs/MiMo-V2.6-RL-harbor-code.