I'm using mat64.Dense and getting weird values when I do element-wise ops in place, like d.Sub(d, b) or…

I'm using mat64.Dense and getting weird values when I do element-wise ops in place, like d.Sub(d, b) or…: a task in MiMo-V2.6-RL-harbor-code: MiMo-V2.6-RL Code (Harbor) (Harbor dataset). Keep the existing public API shape and existing behavior for non-aliasing, non-transposed inputs. The specific…