Okay, so you've heard of PSGD - it's powerful, but it can be slow

What if we could reuse some of the calculations?

That's the goal of the `cache` argument. It speeds things up, making PSGD feasible on even bigger models.