d803bfe2b1
git-subtree-dir: vendor/ruvector git-subtree-split: b64c21726f2bb37286d9ee36a7869fef60cc6900
12 lines
380 B
Rust
12 lines
380 B
Rust
//! # ruvector-attention SDK
|
|
//!
|
|
//! High-level, ergonomic APIs for building attention mechanisms.
|
|
|
|
pub mod builder;
|
|
pub mod pipeline;
|
|
pub mod presets;
|
|
|
|
pub use builder::{flash, multi_head, scaled_dot, AttentionBuilder, AttentionType};
|
|
pub use pipeline::{AttentionPipeline, NormType, PipelineStage};
|
|
pub use presets::{for_graphs, for_large_scale, for_sequences, AttentionPreset};
|