HybridTrainer docstring includes several unimplemented args, like [`device`](https://github.com/traja-team/traja/blob/f87978d02042503ebba57a2f1d1a3b7263a96893/traja/models/train.py#L18). These should be cleaned to reflect the imlemented arguments.
HybridTrainer docstring includes several unimplemented args, like
device. These should be cleaned to reflect the imlemented arguments.