apex

A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch

8,985 stars Python 1 file · ~11,599 tokens

apex MCP Server

MCP server for NVIDIA/apex

Ready-to-Use Client Configurations

{
  "mcpServers": {
    "apex": {
      "command": "npx",
      "args": [
        "-y",
        "@NVIDIA/apex"
      ]
    }
  }
}

{
  "mcpServers": {
    "apex": {
      "command": "npx",
      "args": [
        "-y",
        "@NVIDIA/apex"
      ]
    }
  }
}

{
  "mcpServers": {
    "apex": {
      "command": "npx",
      "args": [
        "-y",
        "@NVIDIA/apex"
      ]
    }
  }
}

{
  "mcpServers": {
    "apex": {
      "command": "npx",
      "args": [
        "-y",
        "@NVIDIA/apex"
      ],
      "disabled": false,
      "autoApprove": []
    }
  }
}

{
  "mcpServers": {
    "apex": {
      "command": "npx",
      "args": [
        "-y",
        "@NVIDIA/apex"
      ]
    }
  }
}

docker run -i --rm mcp/apex