Add DeviceParameters to sweeps

[senecameeks]

I test my changes by modifying these files internally in ~/.virtualenvs//... in the internal repo and then pointing my colab to a local runtime instance.