Angelos Katharopoulos
|
bfdddd644b
|
Expose per-backend availability in C++ and python
|
2025-12-08 15:50:05 -08:00 |
|
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 |
|
Anastasiia Filippova
|
9392fc3f88
|
NCCL backend (#2476)
|
2025-08-21 11:56:15 -07:00 |
|
Cheng
|
828c5f1137
|
Use SmallVector for shapes and strides (#2454)
* Use SmallVector for shapes and strides
* Convert SmallVector to tuple
|
2025-08-05 09:41:03 +09:00 |
|
Anastasiia Filippova
|
515f104926
|
Min / max reductions (#2041)
|
2025-04-09 23:22:20 -07:00 |
|
Angelos Katharopoulos
|
5d68082881
|
Ring docs (#1829)
|
2025-02-28 11:34:21 -08:00 |
|
Angelos Katharopoulos
|
6bf00ef631
|
Fix ring of 2 and allow scalars in API (#1906)
|
2025-02-25 17:03:01 -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 |
|
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 |
|