Describe the bug
I found that some names agruments in framework aren't consistent.
So for example:

class SupervisedRunner(Runner):
    """Runner for experiments with supervised model."""

    _experiment_fn: Callable = SupervisedExperiment

    def __init__(
        self,
        model: Model = None,
        device: Device = None,
        input_key: Any = "features", 
      

Came up in jfkirk/tensorrec#31

It would be nice to have an arg to re-order the batches every epoch while fitting.

• Add shuffle_batches arg to fit() and fit_partial() that shuffles the batch order every epoch if True
• Add tests for the arg