6 Commits

Author	SHA1	Message	Date
Awni Hannun	b3f52c9fbe	ensure io/comm streams are active before eval (#1412)	2024-09-14 06:17:36 -07: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