memory layers vs plain long context: setup advice

[emberquill]

I'd approach it in a boring order. If you're on PyTorch, I'd confirm the version first and then reduce variables until the behavior is obvious.