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_tests.py	Add tile op (#438)	2024-01-12 23:03:16 -08:00
test_array.py	in place ops behave in place, fix some overloads (#411)	2024-01-09 16:05:38 -08:00
test_autograd.py	Add scatter_min VJP (#462)	2024-01-16 00:37:40 -08:00
test_bf16.py	bump pre commit and fix format (#373)	2024-01-04 16:28:52 -08:00
test_blas.py	bump pre commit and fix format (#373)	2024-01-04 16:28:52 -08:00
test_constants.py	feat: Add numpy constants (#428)	2024-01-11 06:47:29 -08:00
test_conv.py	Add isort pre-commit and run (#68)	2023-12-08 11:31:47 -08:00
test_device.py	Add isort pre-commit and run (#68)	2023-12-08 11:31:47 -08:00
test_eval.py	Removes the retain_graph flag (#385)	2024-01-07 15:16:51 -08:00
test_fft.py	Add isort pre-commit and run (#68)	2023-12-08 11:31:47 -08:00
test_graph.py	Multi output primitives (#330)	2024-01-08 16:39:08 -08:00
test_linalg.py	linalg.norm (#187)	2023-12-26 19:42:04 -08:00
test_load.py	GGUF support (#350)	2024-01-10 13:22:48 -08:00
test_losses.py	Additoinal losses (#336)	2024-01-08 14:01:13 -08:00
test_nn.py	Fix BN stats to not expand shape (#409)	2024-01-09 11:54:51 -08:00
test_ops.py	Adds isinf (#445)	2024-01-15 19:50:44 -08:00
test_optimizers.py	Pr template (#99)	2023-12-09 09:36:56 -08:00
test_quantized.py	Allow arbitrary first dimension in quantization kernels. (#458)	2024-01-16 00:46:21 -08:00
test_random.py	make behaviour of dtype arguments consistent and compliant to numpy (#379)	2024-01-05 09:37:46 -08:00
test_reduce.py	bump pre commit and fix format (#373)	2024-01-04 16:28:52 -08:00
test_tree.py	Add isort pre-commit and run (#68)	2023-12-08 11:31:47 -08:00
test_vmap.py	feat: add logicalAnd and logicalOR (#386)	2024-01-08 07:00:05 -08:00