add-memory-prints

star 185

Add detailed memory profiling prints throughout the training framework. Instruments distributed setup, model creation, checkpoint loading, pipeline scheduling, per-layer activations, saved tensor profiling, expert MLP internals, and memory snapshot dumps. Use when user asks to "add memory prints", "instrument memory", "profile memory", "memory breakdown", or "debug memory".

mlc-ai By mlc-ai schedule Updated 6/4/2026

Skill instructions (SKILL.md) could not be loaded from local cache or raw GitHub repository.

Install via CLI
npx skills add https://github.com/mlc-ai/pith-train --skill add-memory-prints
Repository Details
star Stars 185
call_split Forks 13
navigation Branch main
article Path SKILL.md
More from Creator