Module gradient_accumulation 

Modules

utils

Utility functions for gradient accumulation

Structs

GradientAccumulator

Gradient accumulator for micro-batch training

MicroBatchTrainer

Micro-batch trainer that uses gradient accumulation

VariableAccumulator

Variable accumulation scheduler

Enums

AccumulationMode

Gradient accumulation mode

Type Aliases

AdaptiveStepCondition

Type alias for adaptive step conditions