Expand description
zernel optimize โ ML Training optimization tools
Auto-detects and applies optimizations:
- Mixed precision (AMP) with auto-detection of GPU capability
- Smart batch size calculation based on GPU memory and model size
- Gradient checkpointing for memory-constrained training
- Data pipeline bottleneck detection and fix recommendations
- Full optimization scan with one command
Enumsยง
Functionsยง
- run
- run_
auto ๐ - run_
batch_ ๐size - run_
checkpoint ๐ - run_
data_ ๐pipeline - run_
memory ๐ - run_
numa ๐ - run_
precision ๐ - run_
scan ๐