Attention refactor, better categorization and explanation

feat(output): Add token scrubber with per-position top-5 next-token probabilities

Add comprehensive test suite for utility modules (73 tests)