gam 0.3.121

Generalized penalized likelihood engine
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
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65

pub mod analytic_penalties;
pub mod basis;
pub mod construction;
pub mod decoders;
pub mod dictionary;
pub mod geometry;
pub mod latent;
pub mod sae;
pub mod smooth;
pub mod smooth_overrides;
pub mod structure;
pub mod term_builder;
#[path = "smooth/torch_dispatch.rs"]
pub mod torch_dispatch;

pub use analytic_penalties::{
    ARDPenalty, AnalyticPenalty, AnalyticPenaltyKind, AnalyticPenaltyOp, AnalyticPenaltyRegistry,
    BlockOrthogonalityPenalty, BlockSparsityPenalty, DecoderIncoherencePenalty, DifferenceOpKind,
    EdgeRestriction, FrozenAnalyticPenaltyOp, IBPAssignmentPenalty, IsometryDuchonRadialSource,
    IsometryPenalty, IsometryReference, IvaeRidgeMeanGauge, JumpReLUPenalty,
    MechanismSparsityPenalty, NestedPrefixPenalty, NuclearNormPenalty, OrthogonalityPenalty,
    ParametricRowPrecisionPriorPenalty, PenaltyConcavity, PenaltyTier, PsiSlice,
    RowPrecisionPriorPenalty, ScadMcpPenalty, ScalarWeightSchedule, ShapeMonotonicityPenalty,
    SheafConsistencyPenalty, SoftmaxAssignmentSparsityPenalty, SparsityKind, SparsityPenalty,
    TopKActivationPenalty, TotalVariationPenalty, WeightField,
};
pub use basis::matern_gradient::{
    MaternBasisGradientTarget, StreamingMaternBasisGradientEvaluator,
};
pub use decoders::gated_decoder::GatedSAEDecoder;
pub use decoders::interchange_decoder::{
    InterchangeDecodeBackward, InterchangeDecodeForward, InterchangeSwapBackward,
    InterchangeSwapForward, interchange_decode_backward, interchange_decode_forward,
    interchange_swap_backward, interchange_swap_forward,
};
pub use dictionary::{
    LinearDictionaryAssignment, LinearDictionaryConfig, LinearDictionaryFit, fit_linear_dictionary,
};
pub use geometry::PeeledHull;
pub use latent::{
    AuxPriorFamily, AuxPriorStrength, InputLocationDerivative, LatentCoordValues, LatentIdMode,
    LatentManifold,
};
pub use sae::atom_selection::{
    AssignmentSparsityCoupling, AtomLibrary, AtomRecord, AtomSelectionStrategy, EntropicSoftmax,
    L1Relaxed, ShapeRef, TopK,
};
pub use sae::certificates::{
    CriterionCertificate, DirectionalSamples, certificate_from_samples,
    deterministic_probe_direction, probe_step,
};
pub use sae::criterion_atoms::{SaeCriterion, SaeCriterionAtom};
pub use sae::encode::{
    AtlasConfig, AtomEncodeAtlas, BasisHessianLipschitz, CertifiedChart, ChartRegion, EncodeAtlas,
    EncodeResult, KANTOROVICH_THRESHOLD, RowCertificate, row_certificate,
};
pub use sae::manifold::{
    AssignmentMode, CertificateInputs, CurvatureBifurcation, CurvatureWalkReport,
    GumbelTemperatureSchedule, PeriodicHarmonicEvaluator, SaeAssignment, SaeAtomBasisKind,
    SaeBasisEvaluator, SaeManifoldAtom, SaeManifoldLoss, SaeManifoldOuterObjective, SaeManifoldRho,
    SaeManifoldTerm, SaeOuterRhoGradientComponents, ScheduleKind, SphereChartEvaluator,
    TorusHarmonicEvaluator, dictionary_incoherence_report,
    dictionary_incoherence_report_with_dispersion,
};
pub use sae::row_jet_program::{AtomRowBasisJet, RowGate, SaeReconstructionRowProgram};