Allow modification of zero partitioned parameters (#4192)
* Modify zero parameters
* Docs
* py3.6 compatibility
* Update docs
* Update deepspeed/runtime/zero/stage3.py
Co-authored-by: Michael Wyatt <michaelwyatt@microsoft.com>
* Add TODO
* Formatting
---------
Co-authored-by: Logan Adams <114770087+loadams@users.noreply.github.com>
Co-authored-by: Michael Wyatt <michaelwyatt@microsoft.com>
