// General-purpose math utility functions. // Returns a 2x2 rotation matrix. fn rot(a: f32) -> mat2x2 { let c = cos(a); let s = sin(a); return mat2x2(c, s, -s, c); } // Fast approximation of tanh. fn tanh_approx(x: f32) -> f32 { let x2 = x * x; return clamp(x * (27.0 + x2) / (27.0 + 9.0 * x2), -1.0, 1.0); }