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§
- InPlace
Adam - Memory-efficient Adam optimizer with in-place updates
- InPlaceSGD
- Memory-efficient SGD optimizer with in-place updates
Traits§
- InPlace
Optimizer - Trait for in-place parameter updates