mlx/mlx/backend
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
..
accelerate Start to cleanup/unify accelerate and common back-ends (Part 1/N) (#1777) 2025-01-29 14:34:49 -08:00
common Start to cleanup/unify accelerate and common back-ends (Part 1/N) (#1777) 2025-01-29 14:34:49 -08:00
metal Move some kernels to get_template_definition (#1782) 2025-01-21 08:59:44 -08:00
no_cpu Dynamic broadcasting for shapeless compile/export (#1722) 2025-01-09 11:04:24 -08:00
no_metal Dynamic broadcasting for shapeless compile/export (#1722) 2025-01-09 11:04:24 -08:00