Angelos Katharopoulos dfa9f4bc58 An initial quantized matmul implementation (#205) ... * Add quantized matvec * Add quantized matrix matrix with 2nd matrix transposed * Add quantized matmul tests * Add a slow cpu quantized matmul * Add a slightly faster vectorized cpu version		2023-12-18 23:18:57 -08:00
..
mlx_tests.py	added tri / tril / triu (#170)	2023-12-15 17:30:34 -08:00
test_array.py	fix build w/ flatten (#195)	2023-12-17 11:58:45 -08:00
test_autograd.py	Add isort pre-commit and run (#68)	2023-12-08 11:31:47 -08:00
test_bf16.py	Add isort pre-commit and run (#68)	2023-12-08 11:31:47 -08:00
test_blas.py	Add isort pre-commit and run (#68)	2023-12-08 11:31:47 -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	Add isort pre-commit and run (#68)	2023-12-08 11:31:47 -08:00
test_fft.py	Add isort pre-commit and run (#68)	2023-12-08 11:31:47 -08:00
test_load.py	Add isort pre-commit and run (#68)	2023-12-08 11:31:47 -08:00
test_nn.py	Add smoothed L1 loss and enhancements to cross entropy loss (#166)	2023-12-18 07:26:21 -08:00
test_ops.py	Added linspace (#181)	2023-12-18 19:57:55 -08:00
test_optimizers.py	Pr template (#99)	2023-12-09 09:36:56 -08:00
test_quantized.py	An initial quantized matmul implementation (#205)	2023-12-18 23:18:57 -08:00
test_random.py	Add isort pre-commit and run (#68)	2023-12-08 11:31:47 -08:00
test_reduce.py	Add isort pre-commit and run (#68)	2023-12-08 11:31:47 -08:00
test_tree.py	Add isort pre-commit and run (#68)	2023-12-08 11:31:47 -08:00
test_vmap.py	Add move and swap axis, and vmap for slice, concat, and gather (#158)	2023-12-14 12:59:12 -08:00