ConstructionKit
Constants
EXPERT_TYPES = ["transformer", "counter", "bigram"]
TOKENIZER_TYPES = ["char_tokenizer", "bpe_tokenizer"]
WINDOWER_TYPES = ["sequential_window", "sliding_window", "random_window", "dataset"]
Class methods
Extract a ModelConfig from the top-level pipeline graph
Nested types
- ConstructionKit::AddBiasExec
- ConstructionKit::AddExec
- ConstructionKit::AttentionCompoundExec
- ConstructionKit::AttentionExec
- ConstructionKit::BiasParamExec
- ConstructionKit::BroadcastExec
- ConstructionKit::Builder
- ConstructionKit::Compiler
- ConstructionKit::Edge
- ConstructionKit::ElemMulExec
- ConstructionKit::EmbeddingExec
- ConstructionKit::EmbeddingTableExec
- ConstructionKit::ExecutableGraph
- ConstructionKit::ExecutableNode
- ConstructionKit::ExpertInfo
- ConstructionKit::ExpertSpec
- ConstructionKit::FFNExec
- ConstructionKit::GlobalRouterExec
- ConstructionKit::GraphData
- ConstructionKit::GraphDocument
- ConstructionKit::GraphEdge
- ConstructionKit::LayerNormExec
- ConstructionKit::LayerNormFusedExec
- ConstructionKit::LookupExec
- ConstructionKit::LossExec
- ConstructionKit::MatMulExec
- ConstructionKit::MathLossExec
- ConstructionKit::MeanPoolExec
- ConstructionKit::MetricPoint
- ConstructionKit::ModelConfig
- ConstructionKit::ModelSlot
- ConstructionKit::ModelSummary
- ConstructionKit::Node
- ConstructionKit::OutputHeadExec
- ConstructionKit::PortRef
- ConstructionKit::ReLUExec
- ConstructionKit::ScaleParamExec
- ConstructionKit::Server
- ConstructionKit::SoftmaxExec
- ConstructionKit::StepResult
- ConstructionKit::StreamProjExec
- ConstructionKit::Tensor
- ConstructionKit::TrainEvent
- ConstructionKit::TransformerBlockExec
- ConstructionKit::ValidationError
- ConstructionKit::WeightParamExec