About TrainTools
TrainTools is an MCP server published by AparajeetS in the Developer Tools category: recommend paper-backed diagnostics for PyTorch and Hugging Face training problems.
Recent commit activity is the strongest signal that reported issues will get attention.
Install
uvx traintoolspip install traintoolsThis server has no ConduID identity, so agent calls to it are not receipted. Pin the version you install and review the source before granting it credentials.
Tools (11)
**AUMTracker**Which examples look mislabeled by margin dynamics?
Parameters not published.
**BatchInspector**Is this batch broken, imbalanced, or out of scale?
Parameters not published.
**EL2NTracker**Which examples are important or pruneable early in training?
Parameters not published.
**ExampleDynamicsTracker**Which examples are forgotten, hard, ambiguous, or likely mislabeled?
Parameters not published.
**Gradient Noise Scale (GNS)**Is my batch size wasting compute?
Parameters not published.
**GradientAccumulationGNS**Can I get GNS for free during gradient accumulation?
Parameters not published.
**GradientConfusionMonitor**Are micro-batch gradients fighting each other and slowing SGD?
Parameters not published.
**GradientHealthMonitor**Are gradients finite, clipped, vanished, exploded, or too large for the weights?
Parameters not published.
**NeuralCollapseMonitor**Has the classifier entered neural-collapse geometry?
Parameters not published.
**PlasticityProbe**Is my network losing the ability to learn?
Parameters not published.
**TrainGuard**Should I stop training yet?
Parameters not published.
Ask AI
Ask AI about TrainTools
Powered by Claude · Grounded in docs
Security checks
6 checks have not been run yet.