mlx/python
Awni Hannun be98f4ab6b
Reduce a little overhead (#871)
* some small overhead improvements

* use result_type in rms_norm

* remove release force

* fix + use non-vector version

* revert compile change

* fix ops

* a little more overhead

* a little more cleanup and overhead
2024-03-22 17:29:36 -07:00
..
mlx Reduce a little overhead (#871) 2024-03-22 17:29:36 -07:00
src Reduce a little overhead (#871) 2024-03-22 17:29:36 -07:00
tests Set item bug fix (#879) 2024-03-22 12:11:17 -07:00