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?
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?
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?
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 updated202811MITWhy 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 updated1184Why 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.Last updated69MITWhy this server?
Enables scanning of PyTorch checkpoint files (.pt, .pth) for malicious pickle members and other deserialization-based security threats.
AlicenseBqualityCmaintenanceModelSafetyMCP 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?
Supports deployment of PyTorch environments through RunPod's container infrastructure

RunPod MCP Serverofficial
AlicenseCqualityCmaintenanceEnables 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 updated2630658Apache 2.0Why this server?
Enables deployment of pods with PyTorch images, allowing users to specify PyTorch-based container environments.

RunPod MCP Serverofficial
AlicenseCqualityCmaintenanceThis 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 updated3630658Apache 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 updated65