perf: reduce temporary memory when computing attention head results - #1854

perf: reduce temporary memory when computing attention head results#1854

jlarson4 merged 3 commits into

TransformerLensOrg:devTransformerLensOrg/TransformerLens:devfrom

kiteretsu903:perf-bound-head-result-memorykiteretsu903/TransformerLens:perf-bound-head-result-memoryCopy head branch name to clipboard