[Mosaic] Add abs, sign, erf, atan2, reduce_min, reduce_prod support to mgpu#34120

oulgen wants to merge 3 commits intojax-ml:mainjax-ml/jax:mainfrom

oulgen:mgpu-opsoulgen/jax:mgpu-opsCopy head branch name to clipboard