Skip to main content

Module prelude

Module prelude 

Source
Expand description

Convenience re-exports: use burn_mamba::prelude::*; brings the enabled model families and their public types into scope.

Re-exportsยง

pub use crate::mamba1;
pub use crate::mamba2;
pub use crate::mamba3;
pub use crate::modules::CacheStack;
pub use crate::modules::Layer;
pub use crate::modules::Layers;
pub use crate::modules::MambaBidiLayers;
pub use crate::modules::MambaBidiLayersConfig;
pub use crate::modules::MambaBlock;
pub use crate::modules::MambaBlockConfig;
pub use crate::modules::MambaCaches;
pub use crate::modules::MambaLatentNet;
pub use crate::modules::MambaLatentNetConfig;
pub use crate::modules::MambaSsdPath;
pub use crate::modules::MambaVocabNet;
pub use crate::modules::MambaVocabNetConfig;
pub use crate::modules::ResidualsConfig;
pub use crate::modules::StateMoments;
pub use crate::modules::StatePairing;
pub use crate::utils::ClassLatent;
pub use crate::utils::ClassToken;
pub use crate::mamba1::prelude::*;
pub use crate::mamba2::prelude::*;
pub use crate::mamba3::prelude::*;