List of all items
Structs
- attention::multi_head::MultiHeadAttention
- attention::scaled_dot_product::ScaledDotProductAttention
- config::AttentionConfig
- config::AttentionConfigBuilder
- config::GraphAttentionConfig
- config::GraphAttentionConfigBuilder
- config::SparseAttentionConfig
- config::SparseAttentionConfigBuilder
- curvature::ComponentQuantizer
- curvature::FusedCurvatureConfig
- curvature::MixedCurvatureCache
- curvature::MixedCurvatureFusedAttention
- curvature::QuantizationConfig
- curvature::QuantizedVector
- curvature::TangentSpaceConfig
- curvature::TangentSpaceMapper
- graph::dual_space::DualSpaceAttention
- graph::dual_space::DualSpaceConfig
- graph::dual_space::DualSpaceConfigBuilder
- graph::edge_featured::EdgeFeaturedAttention
- graph::edge_featured::EdgeFeaturedConfig
- graph::edge_featured::EdgeFeaturedConfigBuilder
- graph::rope::GraphRoPE
- graph::rope::RoPEConfig
- graph::rope::RoPEConfigBuilder
- hyperbolic::hyperbolic_attention::HyperbolicAttention
- hyperbolic::hyperbolic_attention::HyperbolicAttentionConfig
- hyperbolic::lorentz_cascade::CascadeHead
- hyperbolic::lorentz_cascade::LCAConfig
- hyperbolic::lorentz_cascade::LorentzCascadeAttention
- hyperbolic::mixed_curvature::MixedCurvatureAttention
- hyperbolic::mixed_curvature::MixedCurvatureConfig
- info_bottleneck::DiagonalGaussian
- info_bottleneck::IBConfig
- info_bottleneck::InformationBottleneck
- info_bottleneck::KLDivergence
- info_geometry::FisherConfig
- info_geometry::FisherMetric
- info_geometry::NaturalGradient
- info_geometry::NaturalGradientConfig
- moe::expert::HyperbolicExpert
- moe::expert::LinearExpert
- moe::expert::StandardExpert
- moe::moe_attention::MoEAttention
- moe::moe_attention::MoEConfig
- moe::moe_attention::MoEConfigBuilder
- moe::router::LearnedRouter
- moe::router::TopKRouting
- pde_attention::DiffusionAttention
- pde_attention::DiffusionConfig
- pde_attention::GraphLaplacian
- sdk::builder::AttentionBuilder
- sdk::pipeline::AttentionPipeline
- sparse::flash::FlashAttention
- sparse::linear::LinearAttention
- sparse::local_global::LocalGlobalAttention
- sparse::mask::AttentionMask
- sparse::mask::SparseMaskBuilder
- topology::AttentionPolicy
- topology::PolicyConfig
- topology::TopologyGatedAttention
- topology::TopologyGatedConfig
- topology::WindowCoherence
- training::curriculum::CurriculumScheduler
- training::curriculum::CurriculumStage
- training::curriculum::TemperatureAnnealing
- training::loss::InfoNCELoss
- training::loss::LocalContrastiveLoss
- training::loss::SpectralRegularization
- training::mining::HardNegativeMiner
- training::mining::InBatchMiner
- training::optimizer::Adam
- training::optimizer::AdamW
- training::optimizer::LearningRateScheduler
- training::optimizer::SGD
- traits::EdgeInfo
- traits::Gradients
- traits::SparseMask
- transport::CentroidCache
- transport::CentroidOTAttention
- transport::CentroidOTConfig
- transport::ProjectionCache
- transport::SlicedWassersteinAttention
- transport::SlicedWassersteinConfig
- transport::WindowCache
- unified_report::GeometryReport
- unified_report::MetricValue
- unified_report::ReportBuilder
- unified_report::ReportConfig
Enums
- error::AttentionError
- moe::expert::ExpertType
- pde_attention::LaplacianType
- sdk::builder::AttentionType
- sdk::pipeline::NormType
- sdk::pipeline::PipelineStage
- sdk::presets::AttentionPreset
- sparse::linear::KernelType
- topology::AttentionMode
- topology::CoherenceMetric
- training::curriculum::DecayType
- training::loss::Reduction
- training::mining::MiningStrategy
- unified_report::AttentionRecommendation
- unified_report::MetricType
Traits
- moe::expert::Expert
- moe::router::Router
- training::loss::Loss
- training::mining::NegativeMiner
- training::optimizer::Optimizer
- traits::Attention
- traits::GeometricAttention
- traits::GraphAttention
- traits::SparseAttention
- traits::TrainableAttention
Functions
- hyperbolic::lorentz_cascade::busemann_score
- hyperbolic::lorentz_cascade::einstein_midpoint
- hyperbolic::lorentz_cascade::horosphere_attention_weights
- hyperbolic::lorentz_cascade::lorentz_distance
- hyperbolic::lorentz_cascade::lorentz_inner
- hyperbolic::lorentz_cascade::lorentz_norm_sq
- hyperbolic::lorentz_cascade::project_hyperboloid
- hyperbolic::lorentz_cascade::tangent::exp_map_origin
- hyperbolic::lorentz_cascade::tangent::log_map_origin
- hyperbolic::poincare::exp_map
- hyperbolic::poincare::frechet_mean
- hyperbolic::poincare::log_map
- hyperbolic::poincare::mobius_add
- hyperbolic::poincare::mobius_scalar_mult
- hyperbolic::poincare::poincare_distance
- hyperbolic::poincare::project_to_ball
- sdk::builder::flash
- sdk::builder::multi_head
- sdk::builder::scaled_dot
- sdk::presets::for_graphs
- sdk::presets::for_large_scale
- sdk::presets::for_sequences
- utils::add_vectors
- utils::apply_causal_mask
- utils::apply_dropout
- utils::dot_product
- utils::l2_norm
- utils::masked_softmax
- utils::normalize_vector
- utils::scale_vector
- utils::softmax
- utils::stable_softmax