feat: add W_Q_gram, W_K_gram, QK_alignment weight types to pipeline

New head correlations functionality

optimizing memory transfer

tweaks for performance in production runs; updates to dashboards

handling rare linalg.LinAlgError gracefully

handling rare linalg.LinAlgError gracefully

fixing matmul typo

package directory restructure