Struct LunaEncoder 

pub struct LunaEncoder {
    pub model_cfg: ModelConfig,
    pub device: Device,
    /* private fields */
}

RLX LUNA encoder with per-shape compiled-graph cache.

Fields

model_cfg: ModelConfig

device: Device

Implementations

impl LunaEncoder

pub fn load( config_path: &Path, weights_path: &Path, device: Device, ) -> Result<(Self, f64)>

pub fn describe(&self) -> String

pub fn run_epoch( &mut self, signal: &[f32], chan_pos: &[f32], channel_indices: Option<&[i32]>, n_channels: usize, n_samples: usize, ) -> Result<EpochEmbedding>

Run inference on one epoch.

pub fn run_epoch_opts( &mut self, signal: &[f32], chan_pos: &[f32], channel_indices: Option<&[i32]>, n_channels: usize, n_samples: usize, opts: RunEpochOpts, ) -> Result<EpochEmbedding>

Run inference with explicit options (e.g. skip normalisation for parity vectors).

pub fn run_forward_prepared( &mut self, x_tokenized: &[f32], decoder_queries: &[f32], b: usize, c: usize, t: usize, chan_pos: &[f32], ) -> Result<EpochEmbedding>

Run the RLX graph from prepared x_tokenized / decoder_queries (skips CPU prepare).

pub fn run_rlx_epoch(&mut self, ep: &RlxEpoch) -> Result<EpochEmbedding>

Convenience wrapper around super::io::RlxEpoch.