FlexGEMM Wheels

CUDA 12.8 + PyTorch 2.9.1

Install

pip install flexgemm --find-links https://pozzettiandrea.github.io/flexgemm-wheels/cu128-torch291/

Available Wheels

flex_gemm-0.0.1-cp310-cp310-linux_x86_64.whl
flex_gemm-0.0.1-cp310-cp310-win_amd64.whl
flex_gemm-0.0.1-cp311-cp311-linux_x86_64.whl
flex_gemm-0.0.1-cp311-cp311-win_amd64.whl
flex_gemm-0.0.1-cp312-cp312-linux_x86_64.whl
flex_gemm-0.0.1-cp312-cp312-win_amd64.whl
flex_gemm-0.0.1-cp313-cp313-linux_x86_64.whl
flex_gemm-0.0.1-cp313-cp313-win_amd64.whl