List of all items

Structs

• mamba1::cache::Mamba1Cache
• mamba1::cache::Mamba1CacheConfig
• mamba1::cache::Mamba1Caches
• mamba1::cache::Mamba1CachesConfig
• mamba1::mamba1::Mamba1
• mamba1::mamba1::Mamba1Config
• mamba2::cache::Mamba2Cache
• mamba2::cache::Mamba2CacheConfig
• mamba2::cache::Mamba2Caches
• mamba2::cache::Mamba2CachesConfig
• mamba2::mamba2::Mamba2
• mamba2::mamba2::Mamba2Config
• mamba2::ssd::serial_recalculated::combined_backward::CombinedGrads
• mamba2::ssd::ssd_path::Mamba2SsdInput
• mamba3::double_ssd::cache::Mamba3DoubleSsdCache
• mamba3::double_ssd::cache::Mamba3DoubleSsdCacheConfig
• mamba3::double_ssd::cache::Mamba3DoubleSsdCaches
• mamba3::double_ssd::cache::Mamba3DoubleSsdCachesConfig
• mamba3::double_ssd::double_ssd::step::StepProjection
• mamba3::double_ssd::ssd::serial_recalculated::combined_backward::CombinedGrads
• mamba3::double_ssd::ssd::ssd_path::Mamba3DoubleSsdInput
• mamba3::helpers::TrapezoidCoeffs
• mamba3::mamba3::Mamba3
• mamba3::mamba3::Mamba3Config
• mamba3::moments::Mamba3MomentsInput
• mamba3::moments::backward::MomentsGrads
• mamba3::quat_scan::quat_scan::Quat
• mamba3::single_ssd::cache::Mamba3SingleSsdCache
• mamba3::single_ssd::cache::Mamba3SingleSsdCacheConfig
• mamba3::single_ssd::cache::Mamba3SingleSsdCaches
• mamba3::single_ssd::cache::Mamba3SingleSsdCachesConfig
• mamba3::single_ssd::ssd::serial_recalculated::combined_backward::CombinedSingleSsdGrads
• mamba3::single_ssd::ssd::ssd_path::Mamba3SingleSsdInput
• modules::activation::silu::Silu
• modules::bidi::BidiLayerPair
• modules::bidi::BidiLayers
• modules::bidi::BidiLayersBuilder
• modules::bidi::NoOp
• modules::layer::Layer
• modules::layers::Layers
• modules::layers::LayersBuilder
• modules::loss::bce::BinaryCrossEntropyLoss
• modules::loss::bce::BinaryCrossEntropyLossConfig
• modules::loss::cross_entropy::CrossEntropyLoss
• modules::loss::cross_entropy::CrossEntropyLossConfig
• modules::loss::mse::MseLoss
• modules::multi_gate::MultiGate
• modules::multi_gate::MultiGateResidual
• modules::multi_gate::MultiGateResidualConfig
• modules::network::LatentNetwork
• modules::network::LatentNetworkBuilder
• modules::network::VocabNetwork
• modules::network::VocabNetworkBuilder
• modules::norm::rms_norm::RmsNorm
• modules::norm::rms_norm::RmsNormConfig
• modules::norm::rms_norm_gated::RmsNormGated
• modules::norm::rms_norm_gated::RmsNormGatedConfig
• modules::state_moments::StateMoments
• utils::fprim::F
• utils::fprim::Mask
• utils::scheduler::ConstantLr
• utils::scheduler::CosineAnnealingLr

Enums

• mamba2::ssd::ssd_path::Mamba2SsdPath
• mamba3::cache::Mamba3Cache
• mamba3::cache::Mamba3Caches
• mamba3::moments::recalculated::RotPrim
• mamba3::rotation::RotationKind
• mamba3::rotation::RotationSeq
• mamba3::rotation::RotationState
• mamba3::ssd_path::Mamba3SsdPath
• modules::MambaSsdPath
• modules::bidi::MambaBidiLayers
• modules::bidi::MambaBidiLayersConfig
• modules::bidi::MambaBidiLayersConfigSerde
• modules::bidi::OutputMerge
• modules::bidi::OutputMergeConfig
• modules::bidi::OutputMergeConfigSerde
• modules::cache::MambaCaches
• modules::multi_gate::Residuals
• modules::multi_gate::ResidualsConfig
• modules::multi_gate::ResidualsConfigSerde
• modules::network::MambaLatentNet
• modules::network::MambaLatentNetConfig
• modules::network::MambaLatentNetConfigSerde
• modules::network::MambaVocabNet
• modules::network::MambaVocabNetConfig
• modules::network::MambaVocabNetConfigSerde
• modules::state_moments::StatePairing
• utils::class::ClassLatent
• utils::class::ClassLatentSerde
• utils::class::ClassToken
• utils::class::ClassTokenSerde
• utils::schedule::BidiSchedule
• utils::schedule::Schedule
• utils::scheduler::Lr
• utils::scheduler::LrSerde

Traits

• mamba2::ssd::serial_recalculated::Mamba2AutodiffBackendExt
• mamba2::ssd::serial_recalculated::Mamba2BackendExt
• mamba2::ssd::serial_recalculated::serial_recalculated::Mamba2AutodiffBackendExt
• mamba2::ssd::serial_recalculated::serial_recalculated::Mamba2BackendExt
• mamba3::Mamba3AutodiffBackendExt
• mamba3::Mamba3BackendExt
• mamba3::double_ssd::ssd::serial_recalculated::Mamba3DoubleSsdAutodiffBackendExt
• mamba3::double_ssd::ssd::serial_recalculated::Mamba3DoubleSsdBackendExt
• mamba3::double_ssd::ssd::serial_recalculated::serial_recalculated::Mamba3DoubleSsdAutodiffBackendExt
• mamba3::double_ssd::ssd::serial_recalculated::serial_recalculated::Mamba3DoubleSsdBackendExt
• mamba3::moments::Mamba3MomentsBackendExt
• mamba3::moments::recalculated::Mamba3MomentsBackendExt
• mamba3::quat_scan::quat_scan::Mamba3QuatScanBackendExt
• mamba3::single_ssd::ssd::serial_recalculated::Mamba3SingleSsdAutodiffBackendExt
• mamba3::single_ssd::ssd::serial_recalculated::Mamba3SingleSsdBackendExt
• mamba3::single_ssd::ssd::serial_recalculated::serial_recalculated::Mamba3SingleSsdAutodiffBackendExt
• mamba3::single_ssd::ssd::serial_recalculated::serial_recalculated::Mamba3SingleSsdBackendExt
• modules::MambaBlock
• modules::MambaBlockConfig
• modules::cache::CacheStack
• utils::class::ClassMarker

Macros

• decl_ssd_autodiff_backend_ext
• impl_ssd_backend_ext_for_burn_backends
• utils::class::impl_class_marker

Functions

• mamba2::ssd::serial::k1_ssd_chunk_cumsum
• mamba2::ssd::serial::k2_ssd_bmm
• mamba2::ssd::serial::k3_ssd_chunk_state
• mamba2::ssd::serial::k4_ssd_state_passing
• mamba2::ssd::serial::k5_ssd_chunk_scan
• mamba2::ssd::serial_recalculated::combined_backward::combined_backward
• mamba2::ssd::serial_recalculated::combined_backward::k3_ssd_chunk_state_extended
• mamba2::ssd::serial_recalculated::serial_recalculated::k1_ssd_chunk_cumsum
• mamba2::ssd::serial_recalculated::serial_recalculated::k2_ssd_bmm
• mamba2::ssd::serial_recalculated::serial_recalculated::k3_ssd_chunk_state
• mamba2::ssd::serial_recalculated::serial_recalculated::k4_ssd_state_passing
• mamba2::ssd::serial_recalculated::serial_recalculated::k5_ssd_chunk_scan
• mamba3::double_ssd::ssd::serial::k1_ssd_chunk_cumsum
• mamba3::double_ssd::ssd::serial::k2_ssd_bmm
• mamba3::double_ssd::ssd::serial::k3_ssd_chunk_state
• mamba3::double_ssd::ssd::serial::k4_ssd_state_passing
• mamba3::double_ssd::ssd::serial::k5_ssd_chunk_scan
• mamba3::double_ssd::ssd::serial_recalculated::combined_backward::combined_backward
• mamba3::double_ssd::ssd::serial_recalculated::combined_backward::k3_ssd_chunk_state_extended
• mamba3::double_ssd::ssd::serial_recalculated::serial_recalculated::k1_ssd_chunk_cumsum
• mamba3::double_ssd::ssd::serial_recalculated::serial_recalculated::k2_ssd_bmm
• mamba3::double_ssd::ssd::serial_recalculated::serial_recalculated::k3_ssd_chunk_state
• mamba3::double_ssd::ssd::serial_recalculated::serial_recalculated::k4_ssd_state_passing
• mamba3::double_ssd::ssd::serial_recalculated::serial_recalculated::k5_ssd_chunk_scan
• mamba3::helpers::build_v_with_mimo
• mamba3::helpers::qk_norm_expand_bias
• mamba3::helpers::trapezoidal_coefficients
• mamba3::moments::backward::boundary_carries
• mamba3::moments::backward::d_quat_chunk
• mamba3::moments::backward::d_theta_chunk
• mamba3::moments::backward::moments_phys_bwd
• mamba3::moments::backward::rev_cumsum
• mamba3::moments::backward::scatter_last
• mamba3::moments::recalculated::chunk_cos_sin
• mamba3::moments::recalculated::chunk_decay
• mamba3::moments::recalculated::chunk_mask
• mamba3::moments::recalculated::chunk_states
• mamba3::moments::recalculated::chunk_writes
• mamba3::moments::recalculated::moments_phys_fwd
• mamba3::moments::recalculated::quat_conj_prim
• mamba3::moments::recalculated::quat_mul_prim
• mamba3::moments::recalculated::rotate_chunk
• mamba3::quat_scan::backward::quat_scan_backward
• mamba3::quat_scan::quat_scan::quat_cumprod_recalculated
• mamba3::quat_scan::quat_scan::quat_prefix_product_soa
• mamba3::rotation::quat_conj
• mamba3::rotation::quat_cumprod
• mamba3::rotation::quat_from_scaled_axis
• mamba3::rotation::quat_mul
• mamba3::rotation::quat_normalize
• mamba3::rotation::quat_to_rot4
• mamba3::rotation::rotate_bc_forward
• mamba3::rotation::rotate_bc_step
• mamba3::rotation::rotate_blocks_partial
• mamba3::rotation::rotate_state_rank_blocks
• mamba3::single_ssd::ssd::serial::k5_single_ssd_chunk_scan
• mamba3::single_ssd::ssd::serial_recalculated::combined_backward::combined_backward
• mamba3::single_ssd::ssd::serial_recalculated::serial_recalculated::k5_single_ssd_chunk_scan
• mamba3::step_constant::complex_div
• mamba3::step_constant::complex_mul
• mamba3::step_constant::cos_sin
• mamba3::step_constant::mul_complex_partial
• mamba3::step_constant::mul_quat_partial
• mamba3::step_constant::per_step_angle
• mamba3::step_constant::per_step_generator
• mamba3::step_constant::quat_inv
• mamba3::step_constant::quat_one_minus
• mamba3::step_constant::quat_pow
• mamba3::step_constant::quat_scalar_affine
• modules::activation::log_sigmoid::log_sigmoid
• modules::activation::softplus::softplus
• modules::bidi::bidi_pair_forward
• modules::cache::impl_mamba3::zero_single_ssd_caches
• modules::misc::gqa::gqa_expand_to_heads
• modules::misc::rope::apply_rope
• modules::misc::rope::apply_rope_partial
• modules::misc::rope::wrap_angle
• modules::misc::sanity::sanity
• modules::misc::sanity::sanity_nan
• modules::misc::segsum::segsum
• modules::misc::split::split_into
• utils::class::assert_step_compatible
• utils::class::class_marker_output_indices
• utils::class::class_step_injections
• utils::class::init_class_emb
• utils::class::insert_class_markers
• utils::combined_grad::autodiff_unflatten_pair
• utils::combined_grad::flatten_pair
• utils::combined_grad::unflatten_pair
• utils::div_eps
• utils::fprim::san
• utils::fprim::tri_bool

Constants

• DENY_INF
• DENY_NAN