refactor: rewrite several sections to better match the 1.58 paper & add rope #3

Closed tomsanbear closed 5 months ago