What if one layout could rule all your AI hardware? Researchers from CMU, SJTU, NVIDIA, Princeton, and University of Toronto present AXE Layout. It's a single, simple map that tells data where to go across GPUs, memory, and threads—unifying sharding, tiling, and more.
AXE Layout: Unified AI Hardware Optimization Across GPUs
By
–
Leave a Reply