refactor(tree-attn): simplify tree attention plumbing and restructure attention package#920

garrett4wade merged 1 commit intomaininclusionAI/AReaL:mainfrom

rchardx/cleanupinclusionAI/AReaL:rchardx/cleanupCopy head branch name to clipboard