Angelos Katharopoulos c15fe3e61b Allow arbitrary first dimension in quantization kernels. (#458) ...

* Allow arbitrary first dim on qmm_t and qmv
* Allow arbitrary first dim on qmm and qvm
* Specialized aligned vs unaligned case
* Add more checks for valid quantizations

2024-01-16 00:46:21 -08:00

..

mlx

Add type hint for Module (#412)

2024-01-10 11:23:42 -08:00

src

Adds isinf (#445)

2024-01-15 19:50:44 -08:00

tests

Allow arbitrary first dimension in quantization kernels. (#458)

2024-01-16 00:46:21 -08:00