xma.functionalΒΆ xma.functional.bmm xma.functional.continuous_count xma.functional.cross_entropy xma.functional.fused_linear_cross_entropy xma.functional.fused_residual_add_rmsnorm xma.functional.p_norm xma.functional.rmsnorm xma.functional.sequence_packing xma.functional.softmax xma.functional.swiglu