Cayley TPU

Plugin cayley-tpu in Erlemar/cayley-tpu-plugin. Commands read from the repo's manifest — 1 month ago.

Skills (8)

• cayley-tpu

  Route ambiguous or multi-stage JAX/XLA TPU inference work across the Cayley Tetraminx, Megaminx, cube444, and IHES solvers. Use for TPU search design, capacity planning, preflight, approved launches, monitoring, resume, controlled sweeps, diagnosis, and result verification. Do not use for GPU-only t

• design-static-spmd-search

  Design, review, profile, or change Cayley's pure-JAX static-shape TPU search. Use for V versus all-neighbour Q scoring, owner hashing, per-owner selection, packed all_to_all, deduplication, streamed children, history, no-backtrack filters, exact endgames, symmetry roots, compact records, backpointer

• diagnose-tpu-run

  Diagnose a Cayley JAX TPU run that crashes, stalls, recompiles, exhausts HBM, host RAM, or disk, hangs in collectives, uses unsupported JAX operations, produces partial output, fails replay, or shows unexplained performance or path-quality regression. Use for evidence-backed diagnosis before any fix

• merge-verify-stage

  Collect, normalize, replay-verify, deterministically n-way min-merge, compare, and durably stage Cayley TPU result artifacts. Use for per-rank, per-frame, per-pid, CSV, or JSON outputs; source-set provenance; exact coverage; winner attribution; atomic completion; and preparing a verified artifact wi

• preflight-static-spmd

  Validate a Cayley TPU run immediately before notebook building, upload, launch, or resume, and answer whether a configuration fits. Use for immutable provenance, topology, static-shape divisibility, packed record and backpointer capacity, artifact hashes, expected work units, HBM, host RAM, disk, me

• run-monitor-resume

  Launch, monitor, check, or resume an explicitly approved Cayley Kaggle or GCP TPU inference run. Use when a passing immutable manifest exists and the task involves phase-aware events, provider state, long XLA compilation, durable artifact progress, preemption, partial work, or resuming only missing

• sweep-tpu-configs

  Design, execute, or audit controlled Cayley TPU comparisons of beam width, receive slack, chunks, scorer models, frames, history, endgames, or search policies. Use when a matched control, one-family-at-a-time change, warmup-aware timing, resource margin, append-only failure-inclusive ledger, and rep

• use-jax-tpu-docs

  Resolve current, version-sensitive questions about JAX, jaxlib, XLA, libtpu, shard_map, pjit, sharding, collectives, compilation caching, TPU topology, Kaggle TPU, or Google Cloud TPU. Use when Cayley TPU design or diagnosis needs primary official documentation and precise source attribution.

Similar plugins