mechanistic interpretability: Nonlinear Function