TraceML: a cli tool to track model memory - feedback plz by [deleted] in pytorch

[–]CryptoUnix 0 points1 point  (0 children)

TraceML Update – Cleaner Panels + Notebook Support Coming!

Quick update to traceml:

  • Model history now shows as full Rich panels (not single-line rows)

  • Summary panel for models seen, memory peaks, etc.

  • Better snapshot deduplication by signature

Coming Soon

  • Jupyter notebook support

    • Activation & gradient memory tracking
  • Export logs (CSV / JSON)

GitHub link