Module optimize

Module optimize 

Source
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ยง

OptimizeCommands

Functionsยง

run
run_auto ๐Ÿ”’
run_batch_size ๐Ÿ”’
run_checkpoint ๐Ÿ”’
run_data_pipeline ๐Ÿ”’
run_memory ๐Ÿ”’
run_numa ๐Ÿ”’
run_precision ๐Ÿ”’
run_scan ๐Ÿ”’