Struct GpuAdamW 

pub struct GpuAdamW { /* private fields */ }

GPU AdamW with decoupled weight decay: the decay term is applied to the parameter and never enters the moment estimates.

Implementations

impl GpuAdamW

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 (1-based bias correction input).

pub fn params(&self) -> &AdamParams

Current hyper-parameters.

pub fn set_params(&mut self, params: AdamParams) -> Result<(), GpuOptimError>

Replace the hyper-parameters, validating them first.

Trait Implementations

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

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.