Support running MXNet on YARN - View it on GitHub