xma.layers

• xma.layers.gru
  • GRU
    • GRU.extra_repr()
    • GRU.forward()
    • GRU.reset_parameters()
  • gru()
• xma.layers.linear_attention
  • LinearAttention
    • LinearAttention.forward()
  • linear_attention()
• xma.layers.m2rnn
  • M2RNN
    • M2RNN.forward()
    • M2RNN.reset_parameters()
  • m2rnn()
• xma.layers.moe
  • Experts
    • Experts.down_projection_triton_forward()
    • Experts.extra_repr()
    • Experts.reset_parameters()
    • Experts.torch_forward()
    • Experts.up_projection_triton_forward()
  • MoE
    • MoE.forward()
• xma.layers.rnn
  • RNN
    • RNN.extra_repr()
    • RNN.forward()
    • RNN.reset_parameters()
  • rnn()