mlx/python
Awni Hannun 3dcb286baf
Remove stream from average grads so it uses default (#2532)
* Remove stream from average grads so it uses default

* comment
2025-08-25 15:56:29 -07:00
..
mlx Remove stream from average grads so it uses default (#2532) 2025-08-25 15:56:29 -07:00
scripts nccl dep + default for cuda (#2526) 2025-08-21 17:57:49 -07:00
src allow pathlib.Path to save/load functions (#2541) 2025-08-25 14:58:49 -07:00
tests Remove stream from average grads so it uses default (#2532) 2025-08-25 15:56:29 -07:00