Add all_to_all collective primitive#3164

0xDaizz wants to merge 1 commit intoml-explore:mainml-explore/mlx:mainfrom

0xDaizz:feat/ep-pr1a0xDaizz/mlx:feat/ep-pr1aCopy head branch name to clipboard