Skip to main content

Module memory_efficient

Module memory_efficient 

Source
Expand description

Memory-efficient optimizer implementations

This module provides optimizers designed to minimize memory usage during training, particularly useful for large models or resource-constrained environments.

Structsยง

CircularBuffer
Circular buffer for L-BFGS history with memory management
MemoryConfig
Configuration for memory-efficient optimizers
MemoryEfficientAdam
Memory-efficient Adam optimizer with reduced memory footprint
MemoryEfficientLBFGS
Memory-efficient L-BFGS with improved history management
MemoryEfficientOptimizerBuilder
Builder for memory-efficient optimizers
MemoryPool
Memory pool for reusing tensor allocations