Struct GpuLamb 

pub struct GpuLamb { /* private fields */ }

GPU LAMB with a real layer-wise trust ratio.

One step is two dispatches of the same pipeline: the first advances the moments, materialises the Adam-style update direction and reduces ||params||^2 / ||update||^2 per workgroup; the host finishes the reduction and computes trust = ||params|| / ||update||; the second applies p -= lr * trust * update. When either norm is zero the ratio degenerates to 1, matching the reference implementation.

Implementations

impl GpuLamb

pub fn new(params: AdamParams) -> Result<Self, GpuOptimError>

Create the optimizer with the default WebGPU backend.

pub fn with_config( params: AdamParams, config: GpuOptimizerConfig, ) -> Result<Self, GpuOptimError>

Create the optimizer on an explicit backend.

pub fn backend(&self) -> GpuBackend

Backend actually in use.

pub fn move_to_gpu(&mut self) -> Result<(), GpuOptimError>

Upload the optimizer state to device memory.

Inherent alias for GpuOptimizer::move_to_gpu so callers do not need a turbofish to pin the unused dimension parameter.

pub fn move_to_cpu(&mut self) -> Result<(), GpuOptimError>

Download the optimizer state back to host memory.

Inherent alias for GpuOptimizer::move_to_cpu; see Self::move_to_gpu.

pub fn to_gpu(&mut self) -> Result<(), GpuOptimError>

👎Deprecated since 0.3.2:

renamed to move_to_gpu

Deprecated alias for Self::move_to_gpu.

pub fn to_cpu(&mut self) -> Result<(), GpuOptimError>

👎Deprecated since 0.3.2:

renamed to move_to_cpu

Deprecated alias for Self::move_to_cpu.

pub fn is_gpu_available(&self) -> bool

Whether a real device (not the CPU fallback) is backing this optimizer.

pub fn step_count(&self) -> u64

Number of updates applied so far.

Trait Implementations

impl<D: Dimension> GpuOptimizer<f32, D> for GpuLamb

fn is_gpu_available(&self) -> bool

Check if GPU acceleration is available

fn move_to_gpu(&mut self) -> Result<(), GpuOptimError>

Move optimizer state to GPU

fn move_to_cpu(&mut self) -> Result<(), GpuOptimError>

Move optimizer state back to CPU

fn step_gpu( &mut self, params: &mut Array<f32, D>, gradients: &Array<f32, D>, ) -> Result<(), GpuOptimError>

Perform optimization step on GPU

fn to_gpu(&mut self) -> Result<(), GpuOptimError>

👎Deprecated since 0.3.2:

renamed to move_to_gpu

Deprecated alias for Self::move_to_gpu.

fn to_cpu(&mut self) -> Result<(), GpuOptimError>

👎Deprecated since 0.3.2:

renamed to move_to_cpu

Deprecated alias for Self::move_to_cpu.