3 Commits

Author	SHA1	Message	Date
Anastasiia Filippova	27778156dc	Nccl reduce scatter, all gather (#2727) ... * Added reduce scatter and all gather for nccl * fix unused import, delete unused file * small fix * deleted useless condition * fixed comments * fix bug in eval_gpu, renamed to sum_scatter, fix docs * final fix docs * remove and * Update mlx/distributed/mpi/mpi.cpp Co-authored-by: Awni Hannun <awni.hannun@gmail.com> * fix broken set input output * fixes set output * typo * fix typo * no cpu, no gpu for reduce scatter --------- Co-authored-by: Awni Hannun <awni.hannun@gmail.com>	2025-11-05 08:21:11 -08:00
Awni Hannun	3dcb286baf	Remove stream from average grads so it uses default (#2532) ... * Remove stream from average grads so it uses default * comment	2025-08-25 15:56:29 -07:00
Anastasiia Filippova	9392fc3f88	NCCL backend (#2476)	2025-08-21 11:56:15 -07:00