Module gradient_checkpointing

optirs_core::memory_efficient

Module gradient_checkpointing 

Source
Expand description

Gradient checkpointing for memory optimization

Structs§

AutoCheckpointer
Automatic checkpointing manager for optimization workflows
GradientCheckpointer
Gradient checkpointing manager
MemoryStats
Memory usage statistics
MemoryTracker
Memory usage tracking
MemoryUsage
Memory usage information

Enums§

CheckpointStrategy
Checkpointing strategy for gradient computation