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