ssmall256

1 mods across 1 repository, 2 stars between them.

mlx-metal-kernels

01

ssmall256/mlx-metal-kernels-skill

Skill Claude Code

Provides guidance for writing custom Metal compute kernels using MLX's mx.fast.metalkernel() API for Apple Silicon GPUs (M1, M2, M3, M4). Covers kernel patterns for RMSNorm, LayerNorm, softmax, attention variants (causal, sliding window, GQA), and reductions. Includes profiling, debugging, simdgroupmatrix MMA (M3+)…

not rated 2 8mo ago A 91 tokens original MIT