Gemma Gem

Your personal AI assistant living right inside the browser. Gemma Gem runs Google's Gemma 4 model entirely on-device via WebGPU — no API keys, no cloud, no data leaving your machine. It can read pages, click buttons, fill forms, run JavaScript, and answer questions about any site you visit.

Requirements

• Chrome with WebGPU support
• ~500MB disk for E2B model, ~1.5GB for E4B (cached after first run)

Setup

pnpm install
pnpm build

Load the extension in chrome://extensions (developer mode) from .output/chrome-mv3-dev/.

Usage

1. Navigate to any page
2. Click the gem icon (bottom-right corner) to open the chat
3. Wait for model to load (progress shown on icon + chat)
4. Ask questions about the page or request actions

Architecture

Offscreen Document          Service Worker           Content Script
(Gemma 4 + Agent Loop)  <-> (Message Router)    <-> (Chat UI + DOM Tools)
       |                         |
  WebGPU inference          Screenshot capture
  Token streaming           JS execution

• Offscreen document: Hosts the model via @huggingface/transformers + WebGPU. Runs the agent loop.
• Service worker: Routes messages between content scripts and offscreen document. Handles take_screenshot and run_javascript.
• Content script: Injects gem icon + shadow DOM chat overlay. Executes DOM tools (read_page_content, click_element, type_text, scroll_page).

Tools

Settings

Click the gear icon in the chat header:

• Model: Switch between Gemma 4 E2B (~500MB) and E4B (~1.5GB). Selection persists across sessions.
• Thinking: Toggle native Gemma 4 thinking
• Max iterations: Cap on tool call loops per request
• Shortcuts: Rebind the keyboard shortcuts that toggle and close the chat. Click a shortcut field, press the key combination you want, and it's saved. The ↺ button resets a shortcut to its default. Defaults: Alt+G toggles the chat, Escape closes it. Bindings persist across sessions.
• Clear context: Reset conversation history for the current page
• Disable on this site: Disable the extension per-hostname (persisted)

Keyboard Shortcuts

Both are rebindable from the gear-icon settings panel (see Shortcuts above).

Development

pnpm build              # Development build (with logging, source maps)
pnpm build:prod         # Production build (logging silenced, minified)

Tech Stack

• WXT — Chrome extension framework (Vite-based)
• @huggingface/transformers — Browser ML inference
• marked — Markdown rendering in chat
• Gemma 4 E2B / E4B (onnx-community/gemma-4-E2B-it-ONNX, onnx-community/gemma-4-E4B-it-ONNX) — q4f16 quantization, 128K context

Debugging

All logs are prefixed with [Gemma Gem]. In development builds, info/debug/warn logs are active. Production builds only log errors.

• Service worker logs: chrome://extensions → Gemma Gem → "Inspect views: service worker"
• Offscreen document logs: chrome://extensions → Gemma Gem → "Inspect views: offscreen.html"
• Content script logs: Open DevTools on any page → Console
• All extension pages: chrome://inspect#other lists all inspectable extension contexts (service worker, offscreen document, etc.)

The offscreen document logs are the most useful — they show model loading, prompt construction, token counts, raw model output, and tool execution.

Hardware Requirements (WebGPU)

Estimated minimal requirements (not benchmarked on real devices)

• Integrated GPUs: Intel Xe (Arc), Apple M1+, Qualcomm Adreno work with sufficient shared memory
• Discrete GPUs: Any with 4 GB+ VRAM (e.g. GTX 1650, RX 6500 XT)
• Mobile: iPhone A14+, Snapdragon 8 Gen 1+ (slow)
• Performance: Slow on integrated GPUs, normal on mid-range discrete GPUs, fast on high-end GPUs
• At long contexts (128K), KV cache adds 10-20% memory overhead on top of model weights