Allow user to specify the kubernetes context to use

[tyler-nervana]

The call to load_kube_config should allow the user to specify a context variable: https://github.com/IntelAI/experiments/blob/master/lib/init.py#L9. This could be done through a sensibly named environment variable.