Enum AssignmentMode 

pub enum AssignmentMode {
    Softmax {
        temperature: f64,
        sparsity: f64,
    },
    IBPMap {
        temperature: f64,
        alpha: f64,
        learnable_alpha: bool,
    },
    ThresholdGate {
        temperature: f64,
        threshold: f64,
    },
}

Assignment prior/relaxation used by SaeAssignment.

Variants

Softmax

Row-wise simplex assignment with entropy sparsity.

Fields

temperature: f64

sparsity: f64

IBPMap

Deterministic concrete relaxation of a truncated IBP active set: each atom’s gate is the INDEPENDENT prior-shrunk activation σ(logit/temperature) · π_k, with π_k = (α/(α+1))^{k+1} the stick-breaking prior mean (see ibp_map_row). These are per-atom gates, NOT mixture/simplex responsibilities: they are computed independently per column and do NOT sum to 1 across atoms (there is no row normalization). Each a_k ∈ [0, π_k] ⊂ [0, 1) is the relaxed “atom k is active in this row” indicator of a truncated IBP, not a share of a unit reconstruction budget.

Fields

temperature: f64

alpha: f64

learnable_alpha: bool

ThresholdGate

Hard-thresholded bounded gate: each atom is off (gate = 0) when its logit is at or below threshold, and on with a threshold-centered shifted sigmoid σ((logit − threshold) / temperature) ∈ [0.5, 1) above it.

#1777 RENAMED from JumpReLU (an inaccurate name): this is NOT the literature JumpReLU activation z·1[z>θ], which carries the thresholded MAGNITUDE z. This mode is a thresholded-logistic GATE (a hard-sigmoid gate): it carries no magnitude at all — its output is a bounded [0, 1) indicator. ThresholdGate names it for what it is. It is a member of the gate family (softmax simplex / IBP sigmoid / this hard gate); reconstruction magnitude lives entirely in the decoder curve g_k(t) = φ(t)ᵀ B_k. The discontinuity at threshold (0 → 0.5) is the intended “jump”.

BACK-COMPAT: the constructor Self::threshold_gate is the primary spelling; Self::jumprelu is retained as a deprecated alias, and the FFI string parser accepts both "threshold_gate" and the legacy "jumprelu".

Fields

temperature: f64

threshold: f64

Implementations

impl AssignmentMode

pub fn softmax(temperature: f64) -> Self

pub fn ibp_map(temperature: f64, alpha: f64, learnable_alpha: bool) -> Self

pub fn threshold_gate(temperature: f64, threshold: f64) -> Self

#1777 — construct the hard-sigmoid Self::ThresholdGate (the accurate name for what was jumprelu). Primary spelling; Self::jumprelu is a deprecated alias kept for back-compat.

pub fn jumprelu(temperature: f64, threshold: f64) -> Self

Back-compat alias for Self::threshold_gate (#1777): the mode is a hard-sigmoid gate, not the literature JumpReLU magnitude activation. Retained (NOT #[deprecated], since the workspace denies warnings and many callers and the legacy "jumprelu" FFI token still use it) so existing code keeps compiling; new code should prefer threshold_gate.

pub fn temperature(&self) -> f64

Trait Implementations

impl Clone for AssignmentMode

fn clone(&self) -> AssignmentMode

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Copy for AssignmentMode

impl Debug for AssignmentMode

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.