---
格式版本: 2
标题: "Write Once, Run Everywhere: The Axon DSL for Shape-Safe and Framework-Agnostic LLM Architectures"
原文链接: "https://arxiv.org/abs/2608.19889"
发布日期: "2026-08-20"
发布时间校准状态: "found"
发布时间需复核: "否"
发布时间来源: "rule:local:strict_original_body"
发布时间证据: "**\\[v1\\]** Thu, 20 Aug 2026 10:56:34 UTC (934 KB)"
发布时间校准原因: "规则确认唯一严格发布时间，来源 local:strict_original_body"
发布时间校准置信度: "high"
发布时间候选数量: 18
发布时间严格候选数量: 6
发布时间原页读取状态: "source template page reused from URL open"
发布时间未找到原因: ""
发布时间校准时间: "2026-08-21T16:09:52+08:00"
发布时间仲裁状态: "skipped"
发布时间仲裁尝试次数: 0
发布时间仲裁耗时毫秒: 0
发现时间: "2026-08-21T16:06:59+08:00"
入库时间: "2026-08-21T08:09:52.594Z"
来源平台: "arXiv 学术论文搜索"
搜索渠道: "source_template"
搜索词: "https://arxiv.org/search/?query=Scale-up&searchtype=all"
匹配关键词:
  - "Scale-up"
  - "deployment"
  - "AI"
相关厂家:
  []
相关专家:
  []
内容类型: "网页"
抓取工具: "Free Fetch + Defuddle"
清洗工具: "Defuddle Markdown + Defuddle/Readability 正文提取"
原始附件:
  []
AI优质: "否"
AI打分: 15
AI分档: "非优质"
AI质检状态: "不通过"
AI打分理由: "论文实际内容为Axon DSL语言设计，用于LLM架构移植，与超节点、AI Rack、机柜级AI基础设施、供电散热互连等主题完全无关，仅搜索词命中“Scale-up”，应判无关。"
AI质检模型: "ali-deepseek-v4-flash"
AI质检时间: "2026-08-21T16:10:45+08:00"
AI主题相关性: 0
AI来源权威性: 8
AI新颖性: 5
AI技术细节: 2
AI商业部署信号: 0
AI完整性: 0
AI摘要: "Axon 是一个强类型、Haskell 式语法的领域特定语言，可让 LLM 架构“一次编写、随处运行”，自动编译到 PyTorch、JAX、MLX 和 vLLM 等主流框架。"
AI摘要模型: "ali-deepseek-v4-flash"
AI摘要时间: "2026-09-07T02:12:34.403Z"
采集批次: "2026年8月21日13点55分53秒"
采集批次ID: "20260821-135553-743"
去重键: "https://arxiv.org/abs/2608.19889"
---

## Computer Science > Artificial Intelligence

## Title:Write Once, Run Everywhere: The Axon DSL for Shape-Safe and Framework-Agnostic LLM Architectures

[View PDF](https://arxiv.org/pdf/2608.19889) [HTML (experimental)](https://arxiv.org/html/2608.19889v1)

> Abstract:The entire ecosystem of open-source language models effectively relies on a single platform. What if this platform was forced to shut down tomorrow? Implementing and maintaining efficient model definitions and translating them between different training and inference regimes is a resource-heavy task that severely limits model efficiency and portability, hindering both scaling and deployment. Here, we present Axon, a strongly typed domain-specific language with Haskell-like syntax, that enables a write-once, run everywhere paradigm for LLM architectures. By basing collaboration on a language specification rather than a specific framework's vision, Axon fosters open cooperation and empowers researchers to implement highly specialized architectures without giving up optimization infrastructure or accepting deployment lock-in. Axon allows for concise, auditable specifications that can be automatically compiled to standalone implementations for leading frameworks: PyTorch, PyTorch with Triton, JAX, MLX and vLLM. In 467 inference benchmarking experiments on models ranging from 135M to 32B parameters, we demonstrate median speedups of 7% on PyTorch, 12% on PyTorch with Triton, 91% on JAX, and 107% on MLX, compared to the reference implementations from Transformers. When deployed as native vLLM architectures with PagedAttention and KV-cache, Axon models achieve a 58% median speedup over Transformers implementations.

| Subjects: | Artificial Intelligence (cs.AI); Programming Languages (cs.PL) |
| --- | --- |
| Cite as: | [arXiv:2608.19889](https://arxiv.org/abs/2608.19889) \[cs.AI\] |
|  | (or [arXiv:2608.19889v1](https://arxiv.org/abs/2608.19889v1) \[cs.AI\] for this version) |
|  | [https://doi.org/10.48550/arXiv.2608.19889](https://doi.org/10.48550/arXiv.2608.19889) |

## Submission history

From: Jacob Nielsen \[[view email](https://arxiv.org/show-email/eb9b178b/2608.19889)\]  
**\[v1\]** Thu, 20 Aug 2026 10:56:34 UTC (934 KB)

[Which authors of this paper are endorsers?](https://arxiv.org/auth/show-endorsers/2608.19889) | Disable MathJax ([What is MathJax?](https://info.arxiv.org/help/mathjax.html))