Python Jax

Name: Python Jax
Rating: 0.1 (14 reviews)
Author: jkitchin

MCP Server

by jkitchin · Added 5mo ago

Claude

Install

git clone https://github.com/jkitchin/skillz

Add to Claude

claude mcp add python-jax

Quick Setup Config

Copy & paste ready

~/Library/Application Support/Claude/claude_desktop_config.json (macOS) %APPDATA%\Claude\claude_desktop_config.json (Windows)

{
  "mcpServers": {
    "python-jax": {
      "command": "npx",
      "args": [
        "@modelcontextprotocol/python-jax"
      ]
    }
  }
}

Paste into your Claude Desktop config file, then restart Claude.

About

Expert guidance for JAX (Just After eXecution) - high-performance numerical computing with automatic differentiation, JIT compilation, vectorization, and GPU/TPU acceleration; includes transformations (grad, jit, vmap, pmap), sharp bits, gotchas, and differences from NumPy

Python Jax

Install

Add to Claude

Quick Setup Config

About

Tags

From Our Store

AI Coding Agent Blueprints

Claude Code Power User Kit