AI

Nova: An End-to-End MLIR Compiler for Deep Learning

Researchers have developed an automated compiler called Nova that can optimize the performance of deep learning models on hardware. Unlike traditional tensor frameworks, Nova synthesizes fine-grained kernels directly from the computation's structure, allowing for aggressive whole-graph optimizations and reducing memory-bound overheads. The team has extended Nova to support full Transformer architectures and demonstrated its effectiveness in training a GPT-2 model on Ada 6000
Researchers have developed an automated compiler called Nova that can optimize the performance of deep learning models on hardware. Unlike traditional tensor frameworks, Nova synthesizes fine-grained kernels directly from the computation's structure, allowing for aggressive whole-graph optimizations and reducing memory-bound overheads. The team has extended Nova to support full Transformer architectures and demonstrated its effectiveness in training a GPT-2 model on Ada 6000 GPUs, achieving superior end-to-end throughput compared to traditional eager execution or compilation methods. --- Why it matters: This matters because it enables efficient compilation of large language models (LLMs) on modern hardware, which is crucial for widespread adoption of AI technologies. By reducing memory-bound overheads and enabling aggressive optimizations, Nova has the potential to improve the performance and scalability of various applications that rely on deep learning. Source: https://arxiv.org/abs/2608.00029

This article was originally published at: https://arxiv.org/abs/2608.00029