ssmall256

1 mods across 1 repository, 2 stars between them.

mlx-metal-kernels

01

ssmall256/mlx-metal-kernels-skill

Skill Claude CodeCodex

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+)…

2 6mo ago A 91 tokens original MIT