Sean Naren f7459f5328 DeepSpeed Infinity Update (#7234) ... * Update configs to match latest API * Ensure we move the entire model to device before configure optimizer is called * Add missing param * Expose parameters * Update references, drop local rank as it's now infered from the environment variable * Fix ref * Force install deepspeed 0.3.16 * Add guard for init * Update pytorch_lightning/plugins/training_type/deepspeed.py Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com> * Revert type checking * Install master for CI for testing purposes * Update CI * Fix tests * Add check * Update versions * Set precision * Fix * See if i can force upgrade * Attempt to fix * Drop * Add changelog Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>		2021-06-14 16:38:28 +00:00
..
gpu-tests.yml	DeepSpeed Infinity Update (#7234)	2021-06-14 16:38:28 +00:00
ipu-tests.yml	[IPU] Add special tests for IPUs 2/n (#7833)	2021-06-04 23:23:09 +05:30