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?
Enables deployment of pods with PyTorch images, allowing users to specify PyTorch-based container environments.
AlicenseCqualityBmaintenanceThis 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.Last updated3629155Apache 2.0Why this server?
Supports PyTorch model format for Hailo-8 AI accelerator inference workloads.
AlicenseBqualityCmaintenanceEnables 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.Last updated584MITWhy this server?
Supports deployment of GPU instances using PyTorch/CUDA images as part of the Novita AI platform resource management capabilities.

Novita MCP Serverofficial
AlicenseCqualityCmaintenanceAn 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.Last updated203311MITWhy this server?
Provides diagnostic capabilities specifically for PyTorch workloads, identifying bottlenecks in the forward pass, data movement, and the caching allocator.
AlicenseAqualityBmaintenanceeBPF-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.Last updated1182Why this server?
Supports deployment of PyTorch environments through RunPod's container infrastructure

RunPod MCP Serverofficial
AlicenseCqualityBmaintenanceEnables interaction with the RunPod REST API through Claude or other MCP-compatible clients, providing tools for managing pods, endpoints, templates, network volumes, and container registry authentications.Last updated2629155Apache 2.0Why this server?
Enables creation of distributed PyTorch training workflows with GPU resource management through Flyte's task execution capabilities.
AlicenseBqualityBmaintenanceProvides AI assistants with accurate Flyte V2 knowledge, patterns, and plugin information to help developers write correct Flyte code. It enables tasks like learning the V2 API, finding examples, selecting plugins, and migrating V1 code to V2.Last updated17Apache 2.0Why this server?
Uses PyTorch as the underlying framework for model operations, with optimizations for faster training and reduced memory usage.
FlicenseAqualityCmaintenanceProvides tools for optimizing, fine-tuning, and deploying large language models with Unsloth, enabling 2x faster training with 80% less memory through model loading, fine-tuning, text generation, and model export capabilities.Last updated65Why this server?
Enables scanning of PyTorch checkpoint files (.pt, .pth) for malicious pickle members and other deserialization-based security threats.
AlicenseCqualityCmaintenanceModelSafetyMCP 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.Last updated6MITWhy this server?
Provides access to PyTorch documentation through DevDocs.io, enabling search and retrieval of PyTorch library documentation content.
FlicenseAqualityCmaintenanceProvides access to 600+ documentation libraries from DevDocs.io including Python, JavaScript, React, Django, and more. Enables searching, browsing, and retrieving documentation content directly through Claude Desktop.Last updated53