kernels-community/mlx-quantization-metal-kernels

This is the repository card of kernels-community/mlx-quantization-metal-kernels that has been pushed on the Hub. It was built to be used with the kernels library. This card was automatically generated.

How to use

# make sure `kernels` is installed: `pip install -U kernels`
from kernels import get_kernel

kernel_module = get_kernel("kernels-community/mlx-quantization-metal-kernels")
mxfp4_qmm_n = kernel_module.mxfp4_qmm_n

mxfp4_qmm_n(...)

Available functions

• mxfp4_qmm_n
• mxfp4_qmv
• affine_qmv
• affine_qmm_t
• affine_qmm_n
• affine_qmm_t_nax
• affine_qmm_n_nax
• affine_gather_qmm_rhs_nax

Benchmarks

No benchmark available yet.