rlx-flow 0.2.0

Block assembly-line API for RLX model builders — fusion-first, config-driven
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17

// RLX — versatile ML compiler + runtime.
// Copyright (C) 2026 Eugene Hauptmann, Nataliya Kosmyna.

//! Common imports for model flow authors.

pub use crate::blocks::{
    BertEncoderLayerSpec, BertEncoderLayerStage, BertQkvStyle, CustomStage, EmbedStage,
    GatherAddStage, GatherFromInputStage, GeluFfnStage, LayerNormStage, LinearStage,
    LlamaDecodeLayerSpec, LlamaDecoderSpec, LmHeadStage, NomicEncoderLayerSpec,
    NomicEncoderLayerStage, ResidualAddStage, ResidualSaveStage, RmsNormStage, RopeTablesStage,
    SelfAttnPrefillSpec, SwiGluStage, llama_prefill_layer_composed, llama_prefill_layer_fused,
};
pub use crate::context::{DecodeBindings, FlowState};
pub use crate::{
    BuiltModel, CompileProfile, Emit, FlowStage, FlowValue, LayerStack, ModelFlow, ModelRecipe,
    SideOutputs, WeightSource,
};