Add support for distributed training with Apex

[ryanleary]

Leverage github.com/nvidia/apex to add multi-gpu training