xma.functional.swiglu.pallas_implementation.forward

swiglu_forward_pallas_jit(g: jax.Array, u: jax.Array) → jax.Array

swiglu_forward_pallas_kernel(g_ref, u_ref, y_ref)