E-Graph Flash Attention Optimization

active

Electronic Design Automation Hardware for AI

Using E-graphs to discover optimal ways to balance latency and memory bandwidth trade-offs in flash attention implementations for transformer models.