7 Commits

Author	SHA1	Message	Date
Awni Hannun	058d6ce683	mpi send use input as output (#1750) ... * mpi send use input as output * move earlier	2025-01-06 06:08:43 -08:00
Cheng	0bf19037ca	Remove "using namespace mlx::core" in python/src (#1689)	2024-12-11 15:45:39 -08:00
Angelos Katharopoulos	cdb59faea6	Adds send/recv ops in distributed (#1366)	2024-08-26 23:01:37 -07:00
Awni Hannun	5f7d19d1f5	MPI ops in GPU stream for faster comms (#1356)	2024-08-26 15:12:50 -07:00
Angelos Katharopoulos	0163a8e57a	Add docs for the distributed namespace (#1184)	2024-06-06 11:37:00 -07:00
Angelos Katharopoulos	3de8ce3f3c	In place all-reduce and forgiving init (#1178)	2024-06-03 16:47:47 -07:00
Angelos Katharopoulos	50dfb664db	Comms (#1097) ... * Start the communications branch using MPI * Add ops and primitives * Add python bindings for distributed	2024-05-23 17:04:02 -07:00