Use stateful dataloader to checkpoint data iteration order and token buffer

Use stateful dataloader to checkpoint data iteration order and token buffer #889