DeepSpeed

a149a5a4 - fix the max_out_tokens to use a dynamic range based on available memory