Enum AssignmentMode 

pub enum AssignmentMode {
    Softmax {
        temperature: f64,
        sparsity: f64,
    },
    IBPMap {
        temperature: f64,
        alpha: f64,
        learnable_alpha: bool,
    },
    JumpReLU {
        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.

Fields

temperature: f64

alpha: f64

learnable_alpha: bool

JumpReLU

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. This is NOT literal JumpReLU z·1[z>θ] — the gate carries no magnitude; it is a member of the gate family (softmax simplex / IBP sigmoid / this hard gate) and stays bounded in [0, 1]. 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”.

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 jumprelu(temperature: f64, threshold: f64) -> Self

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.