AI

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