Best PyTorch MCP Servers
PyTorch is an open source machine learning framework based on the Torch library, used for applications such as computer vision and natural language processing.
Why this server?
Allows cleaning of PyTorch model cache (torch hub).
AlicenseAqualityDmaintenanceMCP server for macOS disk cache management. Enables AI agents to free disk space on demand by scanning and clearing developer caches.131Sustainable Use v1.0Why this server?
Supports PyTorch model format for Hailo-8 AI accelerator inference workloads.
AlicenseBqualityAmaintenanceEnables AI assistants to manage homelab infrastructure through automated service installation (Jellyfin, Pi-hole, Ollama, Home Assistant, Frigate NVR), VM operations, AI accelerator support (MemryX, Coral TPU, Hailo-8), and Terraform state management with SSH-based discovery and deployment.584MITWhy this server?
Fetches official PyTorch documentation, including tensor operations and neural network modules.
AlicenseAqualityDmaintenanceEnables managing internal documentation with search and add capabilities, and fetching official documentation from 13+ popular libraries via MCP tools.69MITWhy this server?
Enables deployment of pods with PyTorch images, allowing users to specify PyTorch-based container environments.

RunPod MCP Serverofficial
AlicenseCqualityAmaintenanceThis Model Context Protocol server enables interaction with RunPod's REST API through Claude or other MCP-compatible clients, providing tools for managing pods, endpoints, templates, network volumes, and container registry authentications.364,52974Apache 2.0Why this server?
Supports deployment of GPU instances using PyTorch/CUDA images as part of the Novita AI platform resource management capabilities.

Novita MCP Serverofficial
AlicenseCqualityDmaintenanceAn MCP server that enables seamless management of Novita AI platform resources, currently supporting GPU instance operations (list, create, start, stop, etc.) through compatible clients like Claude Desktop and Cursor.201612MITWhy this server?
Provides diagnostic capabilities specifically for PyTorch workloads, identifying bottlenecks in the forward pass, data movement, and the caching allocator.
AlicenseAqualityAmaintenanceeBPF-based GPU causal observability agent with MCP server. Traces CUDA Runtime and Driver APIs via kernel uprobes and host events via tracepoints to build causal chains explaining GPU latency. 7 tools: get_check, get_trace_stats, get_causal_chains, get_stacks, run_demo, get_test_report, run_sql. Telegraphic compression reduces token usage ~60%. Supports stdio and HTTPS (TLS 1.3) transport.1184Why this server?
Enables scanning of PyTorch checkpoint files (.pt, .pth) for malicious pickle members and other deserialization-based security threats.
AlicenseBqualityDmaintenanceModelSafetyMCP is an MCP server for scanning machine learning model artifacts for unsafe serialization, malicious model patterns, risky packaging, URL-based artifact scanning, and directory-level triage.61MITWhy this server?
Enables creation of distributed PyTorch training workflows with GPU resource management through Flyte's task execution capabilities.
Why this server?
Scans codebases for usage of PyTorch and generates EU AI Act compliance reports.
AlicenseAqualityDmaintenanceScans codebases for AI system usage and generates EU AI Act compliance reports.431MIT