mlx/python
Awni Hannun 881615b072
Faster metal compiled kernels + some fixes (#1486)
* bump mac tests to use py39

* work per thread for compiled kernels

* fixe for large arrays

* fix
2024-10-14 12:45:38 -07:00
..
mlx Add support for grouped 1D convolutions to the nn API (#1444) 2024-09-28 06:41:07 -07:00
src Add mx.random.permutation (#1471) 2024-10-08 19:42:19 -07:00
tests Faster metal compiled kernels + some fixes (#1486) 2024-10-14 12:45:38 -07:00