legume-numeric 0.8.11

Numeric and ML foundation for the legume ecosystem (matrix, Leiden, candle, MCMC)
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25

//! Primitive neural-network building blocks shared across encoders,
//! decoders, and topic models.
//!
//! - [`linear`]: linear variants (incl. non-negative weights, indexed softmax)
//! - [`layers`]: activations, IAF flows, sparsemax, misc layer helpers
//! - [`gcn`]: sparse residual γ-gated GCN block over packed top-K reps
//! - [`batch_norm`]: VarMap-aware BatchNorm (device-transfer safe)

pub mod batch_norm;
pub mod gcn;
pub mod layers;
pub mod linear;
pub mod seed_vars;
pub mod soft_clamp;

pub use batch_norm::{batch_norm, BatchNorm, BatchNormConfig};
pub use gcn::{GcnBlock, SparseEdgeBatch};
pub use layers::{iaf_stack_linear, sparsemax, stack_relu_linear, IAFLayers, StackLayers};
pub use linear::{
    aggregate_linear, aggregate_linear_hard, log_softmax_linear, log_softmax_linear_nobias,
    logsumexp_forward, non_neg_linear, sparsemax_linear, AggregateLinear, NonNegLinear,
    SoftmaxLinear, SparsemaxLinear,
};
pub use seed_vars::seed_uniform_vars;
pub use soft_clamp::{soft_clamp, MASKED_LOGIT_CLAMP};