Name		Name	Last commit message	Last commit date
Latest commit History 2 Commits
.gitattributes		.gitattributes
.gitignore		.gitignore
LICENSE		LICENSE
README.md		README.md
gemm.py		gemm.py
gemm_ggml_bench.py		gemm_ggml_bench.py
gemm_metal.py		gemm_metal.py
gemm_mlx_torch.py		gemm_mlx_torch.py
gemm_tf_bench.py		gemm_tf_bench.py
gemm_tf_coreml_bench.py		gemm_tf_coreml_bench.py
mps_test.py		mps_test.py

Repository files navigation

Matrix Multiplication on Apple Silicon (M1)

Checked solutions

NumPy
TensorFlow ANE https://github.com/tlkh/tf-metal-experiments/blob/main/coreml_matmul.py
TensorFlow Metal https://github.com/tlkh/tf-metal-experiments/blob/main/unified_mem_benchmark.py
Tuned Metal, Torch and Tinygrad https://github.com/tinygrad/tinygrad/blob/master/extra/gemm/metal_matmul.py
Apple's Framework https://github.com/ml-explore/mlx
Python Bindings for GGML https://github.com/abetlen/ggml-python

Useful links

https://github.com/philipturner/metal-benchmarks

About

GEMM on Apple Silicon

benchmark deep-learning gemm applesilicon m1-mac

Report repository

Releases

No releases published

Packages

Languages

Python 100.0%