FLAMIA: FLAsh MIcroscaling Attention for Fast and Causally Safe Training on Blackwell
Introduction Blackwell GPUs execute FP8 and FP4 matrix multiplies 2x and 4x faster than BF16, respectively. To bring that speed to attention, the first problem to solve…