Memory issues with core implementation

[dingo-actual]

While running some tests, I noticed several memory inefficiencies that nullify the advantages of the Infini-Transformer. In particular:

• The loop over segments needs to be moved out of the memory module and into the transformer module.
• The final projection step should be performed on a per-segment basis, rather than for the full sequence.