claude

local-jax-cpu-2b

Serve Gemma 4 E2B under pure JAX on the local CPU — no accelerator, no cloud, no provisioning.

3 on GitHubby xbill9v0.1.0Apache-2.0updated yesterday

Not in a marketplace yet, so it loads for one session with --plugin-dir.

Plugin local-jax-cpu-2b in xbill9/gemma4-dev. Commands read from the repo's manifest — yesterday.

This plugin runs code on your machine — it starts local MCP servers. The exact commands are listed below; read them before installing.

What it installs

Skills (1)

  • local-jax-cpu-2b-management

    Serve Gemma 4 E2B under pure JAX on the local CPU — no accelerator, no cloud, no provisioning. Use when the user asks about running Gemma 4 or JAX locally, serving a model on CPU, starting/stopping the local JAX server, whether a model fits in host RAM, or the local-jax-cpu-2b MCP agent. Triggers in

MCP servers (1)

  • local-jax-cpu-2b

    runs: python3 ${CLAUDE_PLUGIN_ROOT}/skills/local-jax-cpu-2b-management/mcp/server.py

View source on GitHub

More from xbill9/gemma4-dev

All 56 plugins →

Manage AWS Inferentia2 EC2 and Neuron vLLM inference.

2 skills · 1 MCP server

3

Manage Google Cloud TPU capacity and Gemma 4 serving from Rust on Compute Engine TPU VMs: flex-start provisioning, capacity probing, quotas, cost estimates, and SRE diagnostics via the gce-jaxrust-v6e1-2b MCP server. The engine is Rust — rlx's JAX-shaped IR lowered to HLO and executed through libtpu's PJRT plugin, no Python in the serving path. Forked from tpu-jax-v6e1-2b 2026-08-28; nothing provisioned or measured on this path yet.

1 skill · 1 MCP server

3

Compute Engine DevOps for the PyTorch/XLA Gemma 4 E2B rig: flex-start ct6e-standard-1t provisioning, capacity probing, quotas, cost estimates, and SRE diagnostics via the gce-pytorch-v6e1-2b MCP server. Serves google/gemma-4-E2B-it on one TPU v6e chip through public torch_xla (device "xla"), with the xprof + tensorboard profiling kit installed at boot. Forked from tpu-pytorch-v6e1-2b 2026-08-29 and migrated off the Cloud TPU API onto gcloud compute instances.

1 skill · 1 MCP server

3

EC2 G4dn DevOps for the pure-JAX Gemma 4 E2B rig: instance lifecycle, G-family quotas, x86_64 GPU DLAMI resolution, JAX-on-Turing verification, and SRE diagnostics via the gpu-jax-g4dn-2b MCP server. Serves google/gemma-4-E2B-it on an NVIDIA T4 (Turing, SM 7.5, 15360 MiB measured). Forked from gpu-jax-g5g-2b 2026-08-28 as the A/B control that separates Turing from Graviton2, and first served 2026-08-29: decode 13.1 tok/s against the T4G's 13.10, weight bytes identical to the byte, and the xprof profile reproducing 54.4% dtype conversion / 32.8% fp32 GEMV / 0.0% TensorCore — so the sibling's dtype tax is a chip finding, not a host one.

1 skill · 1 MCP server

3

EC2 G5g DevOps for the pure-JAX Gemma 4 E2B rig: instance lifecycle, G-family quotas, arm64 GPU DLAMI resolution, JAX-on-Turing verification, and SRE diagnostics via the gpu-jax-g5g-2b MCP server. Serves google/gemma-4-E2B-it on an NVIDIA T4G (Turing, SM 7.5) behind an AWS Graviton2 host — no build, no CUDA toolkit, no Triton patch, because jax[cuda12] publishes aarch64 wheels that already carry sm_75.

1 skill · 1 MCP server

3

EC2 G6 DevOps for the pure-JAX Gemma 4 E2B rig: instance lifecycle, G-family quotas, x86_64 GPU DLAMI resolution, JAX-on-Ada verification, and SRE diagnostics via the gpu-jax-g6-2b MCP server. Serves google/gemma-4-E2B-it on an NVIDIA L4 (Ada, SM 8.9, 23034 MiB). Forked from gpu-jax-g5g-2b 2026-08-28 to move the same engine off Turing onto a native bfloat16 datapath: measured the same day at 48.3-48.5 tok/s against the T4G's 12.80 at an identical config, with the 54% dtype-conversion tax at 0.0% and decode at the bandwidth roofline.

1 skill · 1 MCP server

3

Similar plugins

career-ops

claudecodex

Open-source AI job search that runs locally in your AI coding CLI: scan job portals, evaluate listings into a structured A-H report with a 1-5 score, tailor your CV, track applications

5 skills · hooks

career-ops-hq/career-ops 82,253

Agent runtimes — local WASM-sandboxed agents (rvagent) + Anthropic Claude Managed Agents (cloud); one interface, local-vs-cloud backends

5 skills · 2 commands · 9 agents

ruvnet/ruflo 81,589

pake

claudecodex

Package any website or local web build into a lightweight desktop app with Pake CLI.

1 skill

tw93/Pake 63,463

Use Alibaba Open Code Review from local Codex.

2 skills

alibaba/open-code-review 50,068

The official CCXT MCP server plus the ccxt-mcp usage skill — market data, balances and opt-in trading across 100+ crypto exchanges and prediction markets, with keys kept local.

1 MCP server

ccxt/ccxt 44,873

Maintain local-jax-cpu-2b?

This page updates itself from your repo. Add the badge to your README, or get it Featured to pin it above the organic results with a followed link to your site.

claude & codex plugins56 plugins