Flashlight: PyTorch Compiler Extensions to Accelerate Attention Variants

Open in new window