Index

_

_cauchy_dot() (ssm.model.block.S4LowRankBlock method) _compute_K() (ssm.model.block.s4_block_interface.S4BlockInterface method) (ssm.model.block.S4BaseBlock method) (ssm.model.block.S4DBlock method) (ssm.model.block.S4LowRankBlock method) _compute_omega() (ssm.model.block.S4LowRankBlock static method) _count_parameters() (ssm.Trainer method) _discretize() (ssm.model.block.S4BaseBlock method) (ssm.model.block.S6Block method)

_discretize_bilinear() (ssm.model.block.S4DBlock method) _discretize_zoh() (ssm.model.block.S4DBlock method) _initialize_ssm_block() (ssm.model.block.MambaBlock method) _preprocess() (ssm.model.block.s4_block_interface.S4BlockInterface static method) (ssm.model.block.S4BaseBlock static method) _recurrent_step() (ssm.model.block.S4BaseBlock static method) (ssm.model.block.S4DBlock static method)

A

argparsing() (ssm.TrainingCLI static method)

C

change_forward() (ssm.model.block.s4_block_interface.S4BlockInterface method) (ssm.model.S4 method) compute_dplr() (in module ssm.utils) compute_hippo() (in module ssm.utils) compute_metrics() (ssm.Trainer method)

compute_S4DInv() (in module ssm.utils) compute_S4DLin() (in module ssm.utils) compute_S4DQuad() (in module ssm.utils) compute_S4DReal() (in module ssm.utils) CopyDataset (class in ssm)

E

EmbeddingBlock (class in ssm.model.block.embedding_block)

F

fit() (ssm.Trainer method) (ssm.TrainingCLI method) forward() (ssm.model.block.embedding_block.EmbeddingBlock method) (ssm.model.block.GatedMLPBlock method) (ssm.model.block.MambaBlock method) (ssm.model.block.residual_block.ResidualBlock method) (ssm.model.block.S6Block method) (ssm.model.Mamba method) (ssm.model.S4 method) (ssm.model.S6 method)

forward_convolutional() (ssm.model.block.s4_block_interface.S4BlockInterface method) (ssm.model.block.S4LowRankBlock method) forward_recurrent() (ssm.model.block.s4_block_interface.S4BlockInterface method)

G

GatedMLPBlock (class in ssm.model.block)

generate_data() (ssm.CopyDataset method)

I

init_model() (ssm.TrainingCLI static method) init_trainer() (ssm.TrainingCLI static method) initialize_A() (ssm.model.block.S4DBlock static method) (ssm.model.block.S4ShiftBlock method)

initialize_dt() (in module ssm.utils) initialize_logger() (ssm.TrainingCLI method)

L

load_config() (ssm.TrainingCLI method)

M

make_selective() (ssm.CopyDataset static method) Mamba (class in ssm.model)

MambaBlock (class in ssm.model.block) model_summary() (ssm.Trainer method) move_to_device() (ssm.Trainer method)

R

ResidualBlock (class in ssm.model.block.residual_block)

S

S4 (class in ssm.model) S4BaseBlock (class in ssm.model.block) S4BlockInterface (class in ssm.model.block.s4_block_interface) S4DBlock (class in ssm.model.block) S4LowRankBlock (class in ssm.model.block)

S4ShiftBlock (class in ssm.model.block) S6 (class in ssm.model) S6Block (class in ssm.model.block) sequential_scan() (ssm.model.block.S6Block static method) set_device() (ssm.Trainer static method)

T

test() (ssm.Trainer method) (ssm.TrainingCLI method)

Trainer (class in ssm) TrainingCLI (class in ssm)

V

vandermonde_matrix() (ssm.model.block.S4DBlock static method)

W

write_on_tensorboard() (ssm.TrainingCLI method)