Make types nices

[will-maclean]

e.g. this is how a OfflineTrainer is created:

let mut trainer = OfflineTrainer::<Adam<<Autodiff<NdArray> as AutodiffBackend>::InnerBackend>, TrainingBacked>::new(...);

Surely there's a more concise way to specify types