mlx/python
Alex Barron a3c287354f
Fast Hadamard Transform (#1249)
* Working hadamard for powers of 2

* working for m*2^k

* add scale and check contiguity

* add size check

* clean up

* fix test

* add grads + vmap

* gpu only

* skip on linux

* test typo

* add cpu impl

* remove gpu only tests

* fix linux build + add is_equivalent
2024-07-09 20:39:01 -07:00
..
mlx CPU binary reduction + Nits (#1242) 2024-06-28 13:50:42 -07:00
src Fast Hadamard Transform (#1249) 2024-07-09 20:39:01 -07:00
tests Fast Hadamard Transform (#1249) 2024-07-09 20:39:01 -07:00