11 Commits

Author	SHA1	Message	Date
Awni Hannun	2a980a76ce	Add stats and limit to common allocator and enable tests (#1988) ... * add stats to common allocator and enable tests * linux memory and default * fix	2025-03-21 12:28:36 -07:00
Awni Hannun	4e1994e9d7	move memory APIs into top level mlx.core (#1982)	2025-03-21 07:25:12 -07:00
Awni Hannun	3c3e558c60	Support transposed head/seq for kv (#1950) ... * support transposed head/seq for kv * fix flaky test * nit	2025-03-10 10:53:45 -07:00
Awni Hannun	c707b2b0a6	Limit compile buffers (#1887) ... * limit compile buffers * maybe not flaky test	2025-02-19 20:28:13 -08:00
Angelos Katharopoulos	0145911bea	Fixes output donation for IO ops on the GPU (#1857)	2025-02-12 10:52:30 -08:00
Angelos Katharopoulos	ccb61d7aae	Ring distributed backend (#1784)	2025-01-27 22:15:01 -08:00
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
Angelos Katharopoulos	914409fef9	Data parallel helper (#1407)	2024-09-16 18:17:21 -07:00
Angelos Katharopoulos	cdb59faea6	Adds send/recv ops in distributed (#1366)	2024-08-26 23:01:37 -07:00
Angelos Katharopoulos	0163a8e57a	Add docs for the distributed namespace (#1184)	2024-06-06 11:37:00 -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