* Harmonize both Trainers API * Fix test * main_prcess -> process_zero
* Documentation for the Trainer API * Address review comments * Address comments