fused-kernels
MLX + Metal implementation of mHC: Manifold-Constrained Hyper-Connections by DeepSeek-AI.
Write the math. Get the kernel. Fused CUDA kernel generation from mathematical specifications.
Fused Triton kernels for normalisation layers (RMSNorm, LayerNorm, GroupNorm, FusedAddRMSNorm)