Awni Hannun 98b6ce3460 Refactor reductions and fix scatter atomics for large sizes (#1300) ...

Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com>

2024-08-22 16:03:31 -07:00

..

mlx

Fix optimizer reloading from checkpoint (#1329)

2024-08-15 07:33:23 -07:00

src

Custom Metal Kernels from Python (#1325)

2024-08-22 13:46:29 -07:00

tests

Refactor reductions and fix scatter atomics for large sizes (#1300)

2024-08-22 16:03:31 -07:00