Searching protocol for "training-metrics"
Write PyTorch docstrings in standard format.
Execute plan batches with checkpoints.