Struct GpuRmsprop 

pub struct GpuRmsprop { /* private fields */ }

GPU RMSprop with optional centering and momentum.

The centered variant keeps a dedicated running mean-gradient buffer (g_avg), which is what makes E[g^2] - E[g]^2 a real variance estimate.

Implementations

impl GpuRmsprop

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

Create the optimizer with the default WebGPU backend.

pub fn with_config( params: RmspropParams, 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 GpuRmsprop

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.