
PyTorch Documentation Search
Supports UIby seanmichaelmcgee
Semantic search prototype for PyTorch documentation with command-line and vector database integration.
What it does
This tool provides a semantic search interface for PyTorch documentation, allowing AI assistants to query technical documentation and code snippets using embeddings for better relevance than simple keyword searches.
Tools
search: Performs a semantic search across the indexed PyTorch documentation to find relevant concepts and code examples.interactive_search: Starts a continuous session for refining queries and exploring PyTorch documentation.
Installation
Add the following to your claude_desktop_config.json:
{
"mcpServers": {
"pytorch-docs": {
"command": "python3",
"args": ["/path/to/pytorch-docs-refactored/scripts/search.py"],
"env": {
"OPENAI_API_KEY": "your_key_here"
}
}
}
}
Supported hosts
- Claude Desktop
Quick install
python3 scripts/search.pyInformation
- Pricing
- free
- Published
- 5/8/2026
- stars





