mlx/python
Awni Hannun 4758c8baa1
Start to cleanup/unify accelerate and common back-ends (Part 1/N) (#1777)
* start to cleanup/unify accelerate and common back-ends

* more progress

* simplify

* add half type and allow infs in simd exp

* unify softmax + quantized, more dispatches to simd quantized mm

* add sin/cos, use simd in vector-scalar ops

* faster CPU vectorize quant

* faster erf/erfinv
2025-01-29 14:34:49 -08:00
..
mlx add dilation for conv 3d layers + test for 3d conv w/ dilation (#1802) 2025-01-28 06:17:07 -08:00
src Start to cleanup/unify accelerate and common back-ends (Part 1/N) (#1777) 2025-01-29 14:34:49 -08:00
tests Start to cleanup/unify accelerate and common back-ends (Part 1/N) (#1777) 2025-01-29 14:34:49 -08:00