legume-numeric 0.9.0

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

//! Primitive neural-network building blocks shared across encoders,
//! decoders, and topic models.
//!
//! - [`linear`]: linear variants (incl. non-negative weights, indexed softmax)
//! - [`layers`]: activations, sparsemax, misc layer helpers
//! - [`batch_norm`](mod@batch_norm): VarMap-aware BatchNorm (device-transfer safe)

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

pub use batch_norm::{batch_norm, BatchNorm, BatchNormConfig};
pub use layers::{sparsemax, stack_relu_linear, 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_declared_vars, seed_uniform_vars};
pub use soft_clamp::{soft_clamp, MASKED_LOGIT_CLAMP};