Struct LieGroupConfig 

pub struct LieGroupConfig {
    pub block_size: usize,
    pub orthogonalize: bool,
    pub target_layers: Vec<String>,
    pub use_cayley: bool,
    pub iterations: usize,
    pub tolerance: f64,
}

Configuration for Lie group optimization

Fields

block_size: usize

Block size for SO(k) decomposition (e.g., 64 or 128)

orthogonalize: bool

Whether to apply orthogonalization

target_layers: Vec<String>

Target layers for optimization (by name pattern)

use_cayley: bool

Use Cayley transform instead of QR

iterations: usize

Number of iterations for optimization

tolerance: f64

Tolerance for convergence

Implementations

impl LieGroupConfig

pub fn new() -> Self

Create a new config with default values

pub fn with_block_size(self, size: usize) -> Self

Set block size for SO(k) decomposition

pub fn with_orthogonalize(self, ortho: bool) -> Self

Enable/disable orthogonalization

pub fn with_target_layers(self, layers: Vec<String>) -> Self

Set target layers

pub fn with_cayley(self, use_cayley: bool) -> Self

Enable/disable Cayley transform

pub fn with_iterations(self, iterations: usize) -> Self

Set number of iterations

pub fn matches_layer(&self, layer_name: &str) -> bool

Check if a layer name matches the target pattern

Trait Implementations

impl Clone for LieGroupConfig

fn clone(&self) -> LieGroupConfig

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for LieGroupConfig

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

Formats the value using the given formatter.

impl Default for LieGroupConfig

fn default() -> Self

Returns the “default value” for a type.