Change dims keyword of sum functions to dim
Change dims keyword of sum functions to dim: a task in MiMo-V2.6-RL-harbor-code: MiMo-V2.6-RL Code (Harbor) (Harbor dataset). For this let $$func(A,x,\tilde x) = A^T(x-\tilde x) = A^Tx - A^T\tilde x = u$$
The task
For this let $$func(A,x,\tilde x) = A^T(x-\tilde x) = A^Tx - A^T\tilde x = u$$
Part of FineEnvs/MiMo-V2.6-RL-harbor-code.