mlx/benchmarks/python
Angelos Katharopoulos e7f5059fe4
Support for quantized matmul with w and w^T (#349)
* Add the metal qvm implementation
* Add qmm_n
* Add gradient wrt to input for quantized_matmul
2024-01-03 14:22:36 -08:00
..
blas Spelling (#342) 2024-01-01 21:08:17 -08:00
comparative Support for quantized matmul with w and w^T (#349) 2024-01-03 14:22:36 -08:00
batch_matmul_bench.py Add isort pre-commit and run (#68) 2023-12-08 11:31:47 -08:00
llama_jax_bench.py copyright + ack 2023-11-30 11:12:53 -08:00
llama_mlx_bench.py copyright + ack 2023-11-30 11:12:53 -08:00
llama_torch_bench.py Add isort pre-commit and run (#68) 2023-12-08 11:31:47 -08:00
single_ops.py Add isort pre-commit and run (#68) 2023-12-08 11:31:47 -08:00
time_utils.py copyright + ack 2023-11-30 11:12:53 -08:00