DeepSpeed

c6370044 - Fix ds_bench, fix args, fix dtypes, fix bw_unit