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

..

blas

Add isort pre-commit and run (#68)

2023-12-08 11:31:47 -08:00

comparative

An initial quantized matmul implementation (#205)

2023-12-18 23:18:57 -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