Skip to main content

Crate ruvector_attention_node

Crate ruvector_attention_node 

Source
Expand description

ruvector-attention-node

Node.js bindings for ruvector-attention via NAPI-RS

This crate provides comprehensive Node.js bindings for:

  • Attention mechanisms (dot-product, multi-head, hyperbolic, flash, linear, local-global, MoE)
  • Training utilities (loss functions, optimizers, schedulers)
  • Async/batch processing
  • Graph attention mechanisms
  • Benchmarking utilities

Re-exports§

pub use attention::AttentionConfig;
pub use attention::DotProductAttention;
pub use attention::FlashAttention;
pub use attention::HyperbolicAttention;
pub use attention::LinearAttention;
pub use attention::LocalGlobalAttention;
pub use attention::MoEAttention;
pub use attention::MoEConfig;
pub use attention::MultiHeadAttention;
pub use training::AdamOptimizer;
pub use training::AdamWOptimizer;
pub use training::CurriculumScheduler;
pub use training::CurriculumStageConfig;
pub use training::DecayType;
pub use training::HardNegativeMiner;
pub use training::InBatchMiner;
pub use training::InfoNCELoss;
pub use training::LearningRateScheduler;
pub use training::LocalContrastiveLoss;
pub use training::LossWithGradients;
pub use training::MiningStrategy;
pub use training::SGDOptimizer;
pub use training::SpectralRegularization;
pub use training::TemperatureAnnealing;
pub use async_ops::AttentionType;
pub use async_ops::BatchConfig;
pub use async_ops::BatchResult;
pub use async_ops::BenchmarkResult;
pub use async_ops::ParallelConfig;
pub use async_ops::StreamProcessor;
pub use graph::DualSpaceAttention;
pub use graph::DualSpaceConfig;
pub use graph::EdgeFeaturedAttention;
pub use graph::EdgeFeaturedConfig;
pub use graph::GraphRoPEAttention;
pub use graph::RoPEConfig;

Modules§

async_ops
NAPI-RS bindings for async and batch operations
attention
NAPI-RS bindings for attention mechanisms
graph
NAPI-RS bindings for graph attention mechanisms
training
NAPI-RS bindings for training utilities

Structs§

LibraryInfo
Library information

Functions§

info
Get library info
version
Get library version