* add NVML binding

* fix flake8

* separate gpu and nvml

* remove stuff

* switch to context

* Clean-up and rename

* fix lint