Expand description
Elementwise SiLU (swish) forward + reverse-mode backward.
Used by hf2q’s ADR-020 Track 1 SwiGLU FFN on GpuTape (iter-11b).
Forward: silu(x) = x · sigmoid(x)
Backward: dx[i] = dy[i] · silu'(x[i])
where silu'(x) = sigmoid(x) · (1 + x · (1 − sigmoid(x)))
Note: mlx-native already has silu_mul_f32 (= silu(gate) * up)
used by the inference forward. This module adds the standalone
silu_f32 + matching backward kernel needed for autograd.
Statics§
Functions§
- dispatch_
silu_ backward_ f32 - Encode
dx[i] = dy[i] · silu'(x[i]).xis the FORWARD INPUT. - dispatch_
silu_ f32 - Encode
output[i] = silu(input[i]) = input[i] · sigmoid(input[i]). - register