Module memory_efficient

Module memory_efficient 

Source

Re-exports§

pub use utils::add_inplace;
pub use utils::apply_inplace;
pub use utils::clip_inplace;
pub use utils::normalize_inplace;
pub use utils::scale_inplace;
pub use utils::subtract_inplace;
pub use adaptive::*;
pub use fused::*;
pub use gradient_checkpointing::*;
pub use mixed_precision::*;

Modules§

adaptive
Dynamic resource adaptation
fused
Fused operations for maximum memory efficiency
gradient_checkpointing
Gradient checkpointing for memory optimization
mixed_precision
Mixed-precision training support
utils
Utility functions for memory-efficient operations

Structs§

InPlaceAdam
Memory-efficient Adam optimizer with in-place updates
InPlaceSGD
Memory-efficient SGD optimizer with in-place updates

Traits§

InPlaceOptimizer
Trait for in-place parameter updates