Coneheads: Hierarchy Aware Attention - Tseng, Albert; Yu, Tao; Liu, Toni; De Sa, Christopher (Advances in Neural Information Processing Systems, 2023)