Disable grokfast and remove gradient checkpointing, add schedule free optimizer, add quadtree position encoder

[lalalune]

This PR removes grokfast, which was causing NaNs, and gradient checkpointing, which optimizes memory but isn't necessary.