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