AI RESEARCH
Nautilus: An Auto-Scheduling Tensor Compiler for Efficient Tiled GPU Kernels
arXiv CS.LG
•
ArXi:2604.14825v1 Announce Type: cross We present Nautilus, a novel tensor compiler that moves toward fully automated math-to-kernel optimization. Nautilus compiles a high-level algebraic specification of tensor operators into efficient tiled GPU kernels. Nautilus's successive lowering design allows high-level optimizations, expression rewrites, and tile optimizations to be jointly applied in a single end-to-end system. Nautilus presents a novel auto-scheduler that discovers sequences of high-level optimizations, while preserving the regular program structure needed by tile optimizers.