Add norm precomputation for SequentialDataset and use it in plain_sgd()