Block

• EmbeddingBlock
  • EmbeddingBlock
    • EmbeddingBlock.forward()
• Gated MLP Block
  • GatedMLPBlock
    • GatedMLPBlock.forward()
• H3Block
• MambaBlock
  • MambaBlock
    • MambaBlock._initialize_ssm_block()
    • MambaBlock.forward()
• ResidualBlock
  • ResidualBlock
    • ResidualBlock.forward()
• S4BlockInterface
  • S4BlockInterface
    • S4BlockInterface._compute_K()
    • S4BlockInterface._preprocess()
    • S4BlockInterface.change_forward()
    • S4BlockInterface.forward_convolutional()
    • S4BlockInterface.forward_recurrent()
• S4BaseBlock
  • S4BaseBlock
    • S4BaseBlock._compute_K()
    • S4BaseBlock._discretize()
    • S4BaseBlock._preprocess()
    • S4BaseBlock._recurrent_step()
• S4LowRankBlock
  • S4LowRankBlock
    • S4LowRankBlock._cauchy_dot()
    • S4LowRankBlock._compute_K()
    • S4LowRankBlock._compute_omega()
    • S4LowRankBlock.forward_convolutional()
• S4DiagonalBlock
  • S4DBlock
    • S4DBlock._compute_K()
    • S4DBlock._discretize_bilinear()
    • S4DBlock._discretize_zoh()
    • S4DBlock._recurrent_step()
    • S4DBlock.initialize_A()
    • S4DBlock.vandermonde_matrix()
• S4ShiftBlock
  • S4ShiftBlock
    • S4ShiftBlock.initialize_A()
• S6Block
  • S6Block
    • S6Block._discretize()
    • S6Block.forward()
    • S6Block.sequential_scan()