zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-12-15 17:39:05 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity
813 Commits 20 Branches 75 Tags
7cbb4aef17889ba8ea1cb3c613696ed76df3c5a6
Commit Graph

5 Commits

Author SHA1 Message Date
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