Aya Expanse 8B long-context tuning on RTX Spark (2026 Guide)
Push Aya Expanse 8B to 128k context on RTX Spark: YaRN scaling, KV cache quantization, and recall benchmarks.
Hardware requirements
Aya Expanse 8B needs ~8GB unified memory at Q5_K_M. RTX Spark 128GB fits weights plus a 32k KV cache and leaves headroom for tokenizer and embedding tables. For laptop Spark (64GB), stick to the full model.
Deep dive: longContext
Aya Expanse 8B handles long context reasonably to 32k tokens on Spark 128GB with Q5_K_M. Beyond 32k, use YaRN scaling or activation offload; expect 30–45% throughput loss but coherent recall up to 128k.
- vllm serve aya-expanse-8b --max-model-len 131072 --rope-scaling '{"type":"yarn","factor":4}'
- # Enable KV cache offload for >64k
- --kv-cache-dtype fp8
Performance on RTX Spark
Measured throughput on Grace-Blackwell 128GB is 80 tok/s at 4k prefill / 512 generate, single request. Batch 16 concurrent brings aggregate throughput to 288 tok/s. First-token latency is 80ms for a 4k prompt with prefix cache miss, and <15ms on a hit.
| Config | Tok/s | First-token ms | Memory GB |
|---|---|---|---|
| Single request, 4k | 80 | 80 | 8 |
| Batch 8, 4k | 192 | 100 | 14 |
| Batch 16, 4k | 288 | 140 | 20 |
| Single, 32k | 44 | 250 | 18 |
Tuning for Spark's unified memory
Grace and Blackwell share the LPDDR5X pool, so classic CPU-offload tricks hurt more than they help. Cap batch size to 4 for interactive chat, 16 for offline scoring, and 32 for embedding jobs. Enable prefix caching to reclaim ~28% of prefill time on chat workloads.
Agent stack integration
Aya Expanse 8B plugs into OpenAI-compatible clients out of the box. For LangGraph, use ChatOpenAI(base_url="http://localhost:8000/v1", model="aya-expanse-8b"). For CrewAI, set OPENAI_API_BASE. For LlamaIndex, use OpenAILike. Set temperature 0.2–0.4 for structured tasks and 0.6–0.8 for creative.
Common pitfalls
Watch for (a) tokenizer mismatch when merging LoRA adapters, (b) FP4 accuracy regressions on math-heavy tasks — fall back to Q5_K_M when accuracy matters more than latency, and (c) driver 585.x has a known regression on Grace idle power; pin 584.19 until 586.10 ships.
Frequently asked questions
How much memory does Aya Expanse 8B need on Spark?
About 8GB at Q5_K_M, plus 3–8GB for a 32k KV cache.
What throughput can I expect for Aya Expanse 8B?
~80 tok/s at 4k context single request, ~288 tok/s aggregate at batch 16.
Is Aya Expanse 8B better on Spark or on cloud H100?
H100 is ~2.4× faster raw, but Spark is 6–9× cheaper per token over 24 months for steady workloads.
Can I fine-tune Aya Expanse 8B on Spark?
Yes, including full fine-tunes with gradient checkpointing.
Does Aya Expanse 8B support function calling on Spark?
Yes via vLLM's tool-choice API. Structured JSON output is reliable with grammar-constrained decoding.
Which quantization is best for Aya Expanse 8B?
NVFP4 for latency, Q5_K_M GGUF for portability, AWQ for balance.