xma.functional.swiglu.pallas_implementation.backward

swiglu_backward_pallas_jit(g: jax.Array, u: jax.Array, dy: jax.Array) → tuple[jax.Array, jax.Array]

swiglu_backward_pallas_kernel(g_ref, u_ref, dy_ref, dg_ref, du_ref)