TrainTools for VS Code

io.github.aparajeets/traintools

Recommend paper-backed diagnostics for PyTorch and Hugging Face training problems.

client:VS Code transport:stdio runtime:pypi

Install TrainTools in VS Code

.vscode/mcp.json

{
  "servers": {
    "traintools": {
      "type": "stdio",
      "command": "uvx",
      "args": [
        "traintools"
      ]
    }
  }
}

Requires agent mode. MCP: Open User Configuration puts the same block in every workspace. VS Code docs