accelerate

f879995c - Should work with deepspeed