mlx/python
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