Added BatchSizeTuner which auto-tunes the batch size for the best performance