Namespace AiDotNet.Diffusion.Memory
Classes
- ActivationPoolStats
Statistics about activation pool usage. Thread-safe counters for concurrent access.
- ActivationPool<T>
Memory pool for tensor activations during diffusion model forward/backward passes.
- DiffusionMemoryConfig
Configuration for diffusion model memory management.
- DiffusionMemoryManager<T>
Memory management utilities for diffusion models including gradient checkpointing, activation pooling, and model sharding integration.
- LayerCheckpointState<T>
State for layer-based gradient checkpointing.
- MemoryEstimate
Memory usage estimate.
- ModelShard<T>
Enables model sharding across multiple devices for large model inference.
- ShardingConfig
Configuration for model sharding.
Interfaces
- IContextualLayer<T>
Interface for layers that accept context (conditioning).
Enums
- ShardingStrategy
Strategy for distributing layers across devices.