xma.functionalΒΆ xma.functional.bmm bmm() xma.functional.continuous_count continuous_count() xma.functional.cross_entropy cross_entropy() xma.functional.fused_residual_add_rmsnorm fused_residual_add_rmsnorm() xma.functional.gru gru() xma.functional.rnn rnn() xma.functional.sequence_packing pack_sequence() unpack_sequence() xma.functional.softmax softmax() xma.functional.swiglu swiglu() xma.functional.fused_linear_cross_entropy fused_linear_cross_entropy() xma.functional.rmsnorm rmsnorm() xma.functional.swiglu_packed swiglu_packed()