Write Once, Run Everywhere: The Axon DSL for Shape-Safe and Framework-Agnostic LLM Architectures
Researchers have developed a new domain-specific language called Axon that allows for the creation of language models that can run on multiple frameworks without significant modifications. This is achieved through a strongly typed syntax and automatic compilation to standalone implementations for popular frameworks such as PyTorch, JAX, and MLX. The authors claim that this approach enables a 'write-once, run everywhere' paradigm for LLM architectures, which could improve mode
Researchers have developed a new domain-specific language called Axon that allows for the creation of language models that can run on multiple frameworks without significant modifications. This is achieved through a strongly typed syntax and automatic compilation to standalone implementations for popular frameworks such as PyTorch, JAX, and MLX. The authors claim that this approach enables a 'write-once, run everywhere' paradigm for LLM architectures, which could improve model efficiency and portability. They demonstrate the effectiveness of Axon through benchmarking experiments on various models.
---
Why it matters: This matters to researchers in AI because it addresses the issue of framework lock-in, where models become tied to a specific platform and cannot be easily transferred or deployed elsewhere. By providing a language-agnostic approach, Axon enables more flexible and efficient model development.
Source: https://arxiv.org/abs/2608.19889
This article was originally published at: https://arxiv.org/abs/2608.19889