Feature: Distributed utilities

There's a number of distributed utilities which are helpful throughout the training code: main_proc_first(), main_proc_only() etc.