feat(model): Add optional memoization to datasets during model training.

feat(model): Add optional memoization to datasets during model training. #724