xma.functional.swiglu.pallas_implementationΒΆ xma.functional.swiglu.pallas_implementation.backward swiglu_backward_pallas_jit() swiglu_backward_pallas_kernel() xma.functional.swiglu.pallas_implementation.forward swiglu_forward_pallas_jit() swiglu_forward_pallas_kernel()