Skip to main content

Crate rlx_bert

Crate rlx_bert 

Source
Expand description

BERT encoder graph builder for RLX.

Lowers a rlx_core::config::BertConfig + weights into an RLX IR graph via the shared rlx_flow::ModelFlow DSL. This crate only builds the graph; compiling and running it (tokenization, device dispatch, pooling) lives in consumer crates such as rlx-embed and rlx-clinicalbert. See the README.

Re-exports§

pub use bert::build_bert_graph_sized;
pub use flow::BertFlow;
pub use flow::build_bert_built;

Modules§

bert
BERT graph builder — lowers a BertConfig + weights into an RLX IR graph.
flow
Tier-0 BERT encoder flow — native ModelFlow assembly.